ConfusionTransposeD
【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Ascend 950PR/Ascend 950DT | √ |
功能说明
- 算子功能：融合reshape和transpose运算。
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 输入张量。 | INT8、INT16、 INT32、 INT64、UINT8、UINT16、UINT32、UINT64、FLOAT16、FLOAT、BFLOAT16 | ND |
| perm | 输入 | 转置后每根轴对应的转置前轴索引。 | INT64 | ND |
| shape | 输入 | reshape后的shape大小。 | INT64 | ND |
| transposeFirst | 输入 | 判断是否先执行transpose操作。 | BOOL | ND |
| y | 输出 | 输出张量。 | 与输入x保持一致。 | ND |
约束说明
-perm张量中元素必须唯一，并在[0,perm的维度数量-1]范围内。 -当transposeFirst为True时，perm的长度必须与x的shape的长度相同，即len(perm)=len(x_shape)；当transposeFirst为False时，perm长度必须与属性输入shape的长度相同，即len(perm)=len(shape)。 -shape中的所有维度乘积必须等于输入张量x的元素总数。
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| aclnn调用 | test_aclnn_confusion_transpose.cpp | 通过aclnnConfusionTranspose接口方式调用ConfusionTransposeD算子。 |
【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库，实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math
创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
