生成对角线张量。如果输入self为一维张量,则返回二维张量,self里元素为对角线值;如果输入self是二维及以上张量,则先进行扁平化(化简为一维张量),再转化为第一种场景处理。
每个算子分为,必须先调用“aclnnDiagFlatGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnDiagFlat”接口执行计算。
[object Object]
[object Object]
参数说明:
[object Object]
[object Object]返回值:
第一段接口完成入参校验,出现以下场景时报错: [object Object]
[object Object]
- 确定性计算:
- aclnnDiagFlat默认确定性实现。
注意:当前接口aclnnDiagFlat暂不具备反向功能。
[object Object]