---
title: permute
description: "permute函数用于张量维度转换。"
url: https://www.hikunpeng.com/document/detail/zh/kunpengboostkithistory/251RC1/accel/kunpengaccel_kpcv_06_0030.html
sourcePath: /source/zh/kunpengboostkithistory/251RC1/accel/kunpengaccel_kpcv_06_0030.html
indexId: 4dd68b3a6bcd719bf1ac4c73075fbafe1da4755a204a2e9668214386029b0f4d77
---
# permute

#### 函数功能

permute函数用于张量维度转换。


#### 函数接口

```
torchvision.ops.permute(dims:List[int]).contiguous()
```


#### 参数

| 参数名 | 描述 | 取值范围 | 输入/输出 |
| --- | --- | --- | --- |
| dims | 序列，指定转置的维度顺序。 | [2, 0, 1] | 输入 |


#### 返回值

- 维度转换后张量。


#### 示例

```
import numpy as np
import torch
from torchvision import transforms, ops
# 创建一个5x5的简单图像 
src = np.array([[1, 0, 0, 0, 0],
               [0, 1, 1, 1, 0],
               [0, 1, 0, 1, 0],
               [0, 1, 1, 1, 0],
               [1, 0, 0, 0, 0]], dtype=np.float32)  

# 将NumPy数组转换为Tensor 
src_tensor = torch.from_numpy(src)  

# 定义维度重排的顺序 dims = [1, 0]  

# 使用torchvision.ops.permute进行图像维度重排 
permute_op = ops.Permute(dims)
permuted_tensor = permute_op(src_tensor)

print(permuted_tensor)
```

运行结果：

```
tensor([[1., 0., 0., 0., 1.],
        [0., 1., 1., 1., 0.],
        [0., 1., 0., 1., 0.],
        [0., 1., 1., 1., 0.],
        [0., 0., 0., 0., 0.]])
```