Блог

У меня есть несколько тензоров факелов со следующими формами

 x1 = torch.Size([1, 512, 177]) x2 = torch.Size([1, 512, 250]) x3 = torch.Size([1, 512, 313])  

Как я могу дополнить все эти тензоры на 0 по последнему измерению, чтобы иметь уникальную форму, такую как ([1, 512, 350]).

Что я пытался сделать, так это преобразовать их в массивы NumPy и использовать эти две строки кода:

 if len(x1) lt; 350:  ff = np.pad(f, [(0, self.max_len - f.shape[0]), ], mode='constant')  f = ff  

Но, к сожалению, это не влияет на последний тусклый цвет, и все равно формы не равны. Любая помощь будет оценена по достоинству Спасибо

Вы можете просто сделать:

 import torch.nn.functional as F  x = F.pad(x, (0, self.max_len - x.size(2)), "constant", 0)