將行重塑為列組

Question

我有許多行向量，我想將它們批處理為列向量並用作Conv1d的輸入。 作為一個例子，我想將張量x重塑為y ，即制作兩組兩個列向量。

# size = [4, 3]
x = torch.tensor([
    [0,  1,  2],
    [3,  4,  5],
    [6,  7,  8],
    [9, 10, 11]
])

# size = [2, 3, 2]
y = torch.tensor([
    [[0,  3],
     [1,  4],
     [2,  5]],
    [[6,  9],
     [7, 10],
     [8, 11]]
])

有沒有辦法只用reshape和類似的功能來做到這一點？ 我能想到的唯一方法是使用循環並復制到一個新的張量中。

Answer 1

您需要使用permute和reshape ：

x.reshape(2, 2, 3).permute(0, 2, 1)
Out[*]:
tensor([[[ 0,  3],
         [ 1,  4],
         [ 2,  5]],

        [[ 6,  9],
         [ 7, 10],
         [ 8, 11]]])

首先，將向量拆分為 2 x.reshape(2,2,3) ，將額外的維度放在中間。 然后使用permute將維度的順序更改為您所期望的。

Answer 2

您也可以使用torch.split和torch.stack類的

torch.stack(x.split(2), dim=2)      # or torch.stack(x.T.split(2, dim=1))
tensor([[[ 0,  3],
         [ 1,  4],
         [ 2,  5]],

        [[ 6,  9],
         [ 7, 10],
         [ 8, 11]]])