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

Question

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

Answer 1

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