pytorch改变tensor维度 示例代码 python

pytorch改变tensor维度 首先使用torch.randn函数创建了一个shape为(2, 3, 4)的Tensor。然后，我们使用view方法将其形状改变为(2, 12)，使用unsqueeze方法在dim=1处增加一个维度，使用squeeze方法在dim=1处删除一个维度，使用permute方法将第1个维度和第2个维度交换。最后，我们分别输出了每个新Tensor的形状。需要注意的是，在实际使用过程中，我们需要根据自己的需求选择合适的操作，并进行维度匹配的检查和调整。