CANN/ops-math转置算子
2026/5/11 12:54:35 网站建设 项目流程

Transpose

【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math

产品支持情况

产品是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品
Atlas 推理系列产品
Atlas 训练系列产品

功能说明

  • 算子功能:对tensor的任意维度进行调换。如输入self是shape为[2, 3, 5]的tensor,dims为(2, 0, 1),则输出是shape为[5, 2, 3]的tensor。

参数说明

参数名输入/输出/属性描述数据类型数据格式
x输入张量需要进行维度置换的输入张量。见下方ND
perm输入张量表示 x 的维度的排列。取值需在[0,self的维度数量-1]范围内。INT64、INT32-
y输出维度最大不超过8维,shape由dims和原self的shape共同决定,dtype需要与self一致。同 xND
  • GPU 支持的数据类型 :数据类型支持 DT_BF16, DT_FLOAT16, DT_FLOAT, DT_DOUBLE, DT_INT64, DT_INT32, DT_UINT8, DT_UINT16, DT_UINT32, DT_UINT64, DT_INT8, DT_INT16, DT_COMPLEX32, DT_COMPLEX64, DT_COMPLEX128, DT_QINT8, DT_QUINT8, DT_QINT16, DT_QUINT16, DT_QINT32, DT_BOOL, DT_HIFLOAT8, DT_FLOAT8_E5M2, DT_FLOAT8_E4M3FN。
  • CPU 支持的数据类型 :数据类型支持同 GPU。

约束说明

调用说明

调用方式样例代码说明
aclnn接口test_aclnn_channel_shuffle通过aclnn_channel_shuffle接口方式调用transpose算子。
aclnn接口test_aclnn_permute通过aclnn_permute接口方式调用transpose算子。

【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询