如何添加张量大小的pytorch

Question

我在使用 PyTorch 时遇到问题。

我正在尝试将张量大小从[1,97,1]增加到[1,97,2] 。 如何在张量大小中添加“1”？

Answer 1

使用垫

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