Diag【免费下载链接】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(展平视为1D)的对角线元素展开为2D对角矩阵计算公式设 $$ \mathbf{x} \in \mathbb{R}^n 输入向量 $$ $$ \mathbf{y} \in \mathbb{R}^{n \times n} 输出对角矩阵 $$其中 $$ \mathbf{y}{i,i} x_i $$ $$ \mathbf{y}{i,j} 0 当 i \ne j $$参数说明参数名输入/输出/属性描述数据类型数据格式x输入公式中的x。INT64、INT32、FLOAT、FLOAT16、DOUBLE、BF16、COMPLEX64、COMPLEX128NDy输出公式中的y。INT64、INT32、FLOAT、FLOAT16、DOUBLE、BF16、COMPLEX64、COMPLEX128NDAtlas 训练系列产品 、 Atlas 推理系列产品 、 Atlas 200I/500 A2 推理产品 、 Atlas A2 训练系列产品/Atlas A2 推理系列产品 、 Atlas A3 训练系列产品/Atlas A3 推理系列产品 不支持BF16。约束说明只支持输入shape维度1-4维对应输出2-8维不支持标量输入。调用说明调用方式调用样例说明图模式调用NA通过算子IR构图方式调用diag算子【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考