如何添加张量大小火炬

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常量。