torch tensor对调
时间: 2023-10-11 21:06:02 浏览: 91
详解torch.Tensor的4种乘法
5星 · 资源好评率100%
要交换张量的维度顺序,可以使用PyTorch中的`permute`方法。下面是一个示例,展示如何对调张量的维度:
```python
import torch
# 创建一个形状为(3, 4, 5)的张量
tensor = torch.randn(3, 4, 5)
# 使用permute交换维度顺序,对调第1维和第2维
transposed_tensor = tensor.permute(1, 0, 2)
print(transposed_tensor.shape) # 输出: torch.Size([4, 3, 5])
```
在上述示例中,我们创建了一个形状为(3, 4, 5)的张量,并使用`permute`方法将第1维和第2维进行了对调。最终,`transposed_tensor`的形状为(4, 3, 5)。
阅读全文