
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、BFLOAT16NDperm输入转置后每根轴对应的转置前轴索引。INT64NDshape输入reshape后的shape大小。INT64NDtransposeFirst输入判断是否先执行transpose操作。BOOLNDy输出输出张量。与输入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),仅供参考