transpose

函数功能

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

功能仅支持在Host侧执行操作,如需处理Device侧Tensor数据,请先通过to_host接口,将Device侧数据转移到Host侧,再进行转置。

函数原型

1
transpose(input: Tensor, axes: List)

输入参数说明

参数名

类型

说明

input

Tensor类

待转置Tensor类。

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

axes

List[int]

转置选项,默认值为空。

如果未指定具体axes,则默认生成反序“axes”“input”中的数据进行反序转置。

例如:三维张量默认生成反序“axes”为{2, 1, 0} 。

返回参数说明

返回转置后的Tensor类数据。

支持的型号

Atlas 200/300/500 推理产品

Atlas 200I/500 A2 推理产品

Atlas 推理系列产品