在最后一个暗淡处填充多个火炬张量答案

【问题标题】：Pad multiple torch tensor over the last dim在最后一个暗淡处填充多个火炬张量
【发布时间】：2021-10-21 22:12:36
【问题描述】：

我有多个具有以下形状的火炬张量

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) < 350:
            ff = np.pad(f, [(0, self.max_len - f.shape[0]), ], mode='constant')
            f = ff

但不幸的是，它不会影响最后的暗淡，而且形状仍然不相等。任何帮助将不胜感激谢谢

【问题讨论】：

【解决方案1】：

你可以这样做：

import torch.nn.functional as F

x = F.pad(x, (0, self.max_len - x.size(2)), "constant", 0)

【讨论】：