CANN/ops-math 共轭转置算子 ConjugateTranspose【免费下载链接】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 训练系列产品√功能说明算子功能按perm指定的维度顺序对输入x做转置并对结果取共轭。对实数类型共轭等价于恒等等价于普通Transpose。计算公式y conj(transpose(x, perm))其中y的第i维大小为x的第perm[i]维大小。参数说明参数名输入/输出/属性描述数据类型数据格式x输入待转置的输入张量维度需大于1。COMPLEX64、COMPLEX128、FLOAT16、FLOAT、DOUBLE、BOOL、INT8、INT16、INT32、INT64、UINT8、UINT16、UINT32、UINT64NDperm输入1-D张量描述维度置换顺序元素个数等于x的维数。INT32、INT64NDy输出转置并取共轭后的张量类型与x一致。与x相同ND约束说明x的维数需大于1且不超过7维。perm必须为1-D元素个数等于x的维数取值为x各维度的一个排列。调用说明调用方式样例代码说明图模式调用test_geir_conjugate_transpose通过算子IR构图方式调用conjugate_transpose算子。【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考