连接火炬张量

Question

我在 PyTorch 中有两个张量：

a.shape, b.shape
# (torch.Size([512, 28, 2]), torch.Size([512, 28, 26]))

我的目标是 join/merge/concatenate 它们在一起，这样我就得到了形状：(512, 28, 28).

我试过了：

torch.stack((a, b), dim = 2).shape
torch.cat((a, b)).shape

但其中 none 似乎有效。

我使用的是 PyTorch 版本：1.11.0 和 Python 3.9.

帮忙？

Answer 1

将 dim 参数设置为 2 以连接最后一个维度：

a = torch.randn(512, 28, 2)
b = torch.randn(512, 28, 26)

print(a.size(), b.size())

# set dim=2 to concat over 2nd dimension
c = torch.cat((a, b), dim=2)

print(c.size())

torch.Size([512, 28, 2]) torch.Size([512, 28, 26])
torch.Size([512, 28, 28])

连接火炬张量

Concatenate torch tensors

multidimensional-array

pytorch

tensor