重塑 PyTorch 张量，使矩阵是水平的

Question

我正在尝试将形状 (n, i, j) 的 3 维 PyTorch 张量中的 n 矩阵组合成形状 的单个二维矩阵（i，j*n）。这是一个简单的例子，其中 n=2, i=2, j=2:

m = torch.tensor([[[2, 3],
                   [5, 7]],
                  [[11, 13],
                   [17, 19]]])
m.reshape(2, 4)

我希望这会产生：

tensor([[ 2,  3, 11, 13],
        [ 5,  7, 17, 19]])

但它却产生：

tensor([[ 2,  3,  5,  7],
        [11, 13, 17, 19]])

我该怎么做？我尝试了 torch.cat 和 torch.stack，但它们需要张量元组。我可以尝试创建元组，但这似乎效率低下。有更好的办法吗？

Answer 1

要将 n + j 与 reshape 结合起来，您需要它们在形状上是一致的。可以使用 swapaxes 修复它：

m = torch.tensor([[[2, 3],
               [5, 7]],
              [[11, 13],
               [17, 19]]])
m=m.swapaxes( 0,1 ) 
m.reshape(2, 4)

---

tensor([[ 2,  3, 11, 13],
        [ 5,  7, 17, 19]])