二上的割炬张量尺寸

Question

如果我有一个形状为[2，12]的火炬张量，是否可以用它来制作形状为[3，2，4]的张量，这样数据就可以沿着最后一个维度分割成块？View函数不会更改数据的顺序，但这里我需要更改顺序，即新维度将是第一个，第二个维度不会更改

以下是数据必须如何拆分的示例图像

[2, 12]  
............   
............  

[3, 2, 4]  
....|....|....   
....|....|....   

torch.split函数在形状上正是我所需要的，但它输出的不是张量，而是元组

Answer 1

我假设您希望将12大小的维度拆分为前导的3个维度和尾随的4个维度。您只需查看和排列尺寸：

X = torch.permute(X.view(2, 3, 4), (1, 0, 2))