可通过指定的一组轴维度(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 推理系列产品