如何连接具有 2 个不同维度的 2d 张量

【问题标题】：How to concatenate 2d tensors with 2 different dimensions如何连接具有 2 个不同维度的 2d 张量
【发布时间】：2021-04-04 13:14:29
【问题描述】：

我想知道是否可以连接两个具有不同形状的不同 pytorch 张量。一个张量的形状为 torch.Size([247, 247])，另一个张量的形状为 torch.Size([10, 183])。是否可以在 dim=1 上使用 torch.cat() 连接这些？

【问题讨论】：

【解决方案1】：

我认为你应该使用广播。也就是说，沿维度 0（达到 247）广播 torch.Size([10, 183]) 或对其他维度进行广播。要使 torch.cat 正常工作，您需要有匹配的维度来尝试连接。

【讨论】：