pytorch - 将 1 x 1 x 33 张量转换为 1 x 1 x 34，同时保持值并保留第 34 个索引 0

Question

我有一个 0 和一个 1 在 1 x 1 x 33 张量中的张量，我希望所有为 1 的索引都保持为 1，但在 1 x 1 x 34 张量中。这个的标准做法是什么？显然为此使用“重塑是不好的”。

标签： pytorch

Answer 1

您可以使用pad零填充张量的开头或结尾。

例如

import torch.nn.functional as F
x = F.pad(x, (0, 1))

x将用 1 个零填充 3d 张量的最后一个维度的末尾。