连接两个张量

【问题标题】：Concat two tensors连接两个张量
【发布时间】：2021-11-20 05:54:28
【问题描述】：

我想连接两个大小为 a: torch.Size([16, 1]) 和 b: torch.Size([16, 120]) 的张量大小为torch.Size([16, 121])

你能帮忙吗？

【问题讨论】：

这能回答你的问题吗？ Concat tensors in PyTorch
不是真的，我试过了，但对我不起作用

标签： pytorch concatenation tensor

【解决方案1】：

这里可以使用torch.cat()函数。

例子：

>>> a = torch.rand([16,1])
>>> b = torch.rand([16,120])
>>> a.size()
torch.Size([16, 1])
>>> b.size()
torch.Size([16, 120])
>>> c = torch.cat((a,b),dim=1)
>>> c.size()
torch.Size([16, 121])

您要做的是连接第一维 (dim=1) 上的张量。

【讨论】：

如果我希望得到的张量是 torch.Size([16, 1]) 怎么办？
@LamaMo 你的目的是什么？
@LamaMo，这两个张量是什么意思？您希望结果张量表示什么。我认为这将决定如何去做。