transpose_operator

可通过指定的一组轴维度（axes），对输入的Tensor数据进行转置处理，如未指定具体轴维度，则默认对Tensor数据进行反序转置。

功能仅支持在Device侧的Tensor数据。

transpose_opeartor(input: Tensor, axes: List[int])

参数名	类型	说明
input	Tensor类	待转置Tensor类。维度支持2维、3维、4维，数据类型支持base.dtype.float32、base.dtype.float16、base.dtype.uint8，具体请参见Tensor类的属性列表。
axes	List[int]	转置选项，长度必须与输入Tensor的维度一致。如传入空列表，则按照反序转置进行处理。若不为空列表则axes中的元素必须包含[0, size-1]区间中的数且每个数只能出现一次，其中size为输入Tensor的维度。

参数名

类型

说明

input

Tensor类

待转置Tensor类。

维度支持2维、3维、4维，数据类型支持base.dtype.float32、base.dtype.float16、base.dtype.uint8，具体请参见Tensor类的属性列表。

axes

List[int]

转置选项，长度必须与输入Tensor的维度一致。

返回转置后的Tensor类数据。

父主题： TensorOperations