как добавить тензор размера pytorch

Question

У меня проблема при работе с PyTorch.

Я пытаюсь увеличить размер тензора с [1,97,1] до [1,97,2]. Как я могу добавить «1» в тензорном размере?

Answer 1

Используйте pad

import torch.nn.functional as F
a = torch.empty((1,97,1))

a = F.pad(input=a, pad=(0,1))     # pad = (padding_left,padding_right)

print(a.shape)
>>> torch.Size([1, 97, 2])

. По умолчанию он будет дополняться константой 0 по умолчанию.