Как уравнять размеры двух тензоров размеров [64, 3, 128, 128] и [64, 64, 127, 127]

Question

Я пытаюсь добавить два тензора измерения [64, 64, 127, 127] и [64, 3, 127, 127] в pytorch, но мне нужно добавить нули во второй тензор, чтобы приравнять их размеры к [64, 64, 127, 127].

Я пытался добавить нули через нарезку, но это не надежно

Answer 1

Первый легко визуализировать:

import torch
zeros = torch.zeros(2, 2, 2, 10)
ones = torch.ones(2, 2, 2, 5)

zeros[:, :, :, :5] = ones
print(ones)
print(zeros)

Второй - то, что вам нужно (нули и единицы - только для моего теста)

ones = torch.ones(64, 3, 127, 127)
zeros = torch.zeros(64, 61, 127, 127)
zeros[:, :3, :, :] = ones

ИЛИ с помощью cat()Функция проста, как это:

torch.cat((ones, zeros), 1)