文件最后提交记录最后更新时间
6 个月前
2 个月前
6 个月前
6 个月前
4 个月前
README

Trace

产品支持情况

产品 是否支持
Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件

功能说明

  • 算子功能:计算矩阵对角线元素的和。
  • 计算公式:

y=∑ixiiy = \sum_{i} x_{ii}

参数说明

参数名 输入/输出 描述 数据类型 数据格式
x 输入 输入矩阵 见下方 ND
y 输出 输出标量 见下方 ND
  • Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件:数据类型支持FLOAT、FLOAT16。

约束说明

调用说明

调用方式 样例代码 说明
aclnn接口 test_trace 通过[aclnnTrace]接口方式调用trace算子。