CANN/ops-math 融合转置D算子
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),仅供参考
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2598136.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!