如何添加张量大小的 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。