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

Posted 2023-03-16

【中文标题】如何连接具有 2 个不同维度的 2d 张量

【英文标题】：How to concatenate 2d tensors with 2 different dimensions

【发布时间】：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 正常工作，您需要有匹配的维度来尝试连接。