如何連接具有 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 正常工作，您需要具有匹配的維度，以便您嘗試連接。