文件最后提交记录最后更新时间
2 天前
4 天前
4 天前
10 天前
16 天前
1 个月前
1 天前
README

MatrixSetDiagV2

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品 ×
Atlas A2 训练系列产品/Atlas A2 推理系列产品 ×
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品 ×
Atlas 训练系列产品 ×

功能说明

  • 算子功能:将输入tensor的对角线元素替换为对角线tensor的值。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
input 输入 待进行替换的原始tensor。 FLOAT、FLOAT16、BFLOAT16、INT8、UINT8、INT16、UINT16、INT32、UINT32、INT64、UINT64、BOOL、COMPLEX64、DOUBLE ND
diagonal 输入 对角线tensor。 与input的数据类型保持一致。 ND
k 输入 对角线的取值范围。 INT32。 ND
output 输出 进行替换后的tensor。 与input的数据类型保持一致。 ND

约束说明

  • input的维度最少为2维,最大不超过8维。
  • k是数据类型INT32的标量或长度为2的向量。
  • k为长度2的向量时,需满足k[1] >= k[0]
  • k为标量或k[0] = k[1](单对角线)时,diagonal的维度比input小1维:最后一维的长度为maxDiagLen = min(row + min(k[1], 0), col - max(k[0], 0)),其中rowcol分别为input最后两维的长度;其余维度与input的对应维度一一相等。
  • k不是标量且k[0] != k[1](多对角线)时,diagonal的维度与input相同:最后两维的长度分别为numDiags = k[1] - k[0] + 1maxDiagLen = min(row + min(k[1], 0), col - max(k[0], 0)),其中rowcol分别为input最后两维的长度,numDiags为对角线数量;除最后两维外,其余维度与input的对应维度一一相等。

调用说明

调用方式 样例代码 说明
图模式调用 test_geir_matrix_set_diag_v2 通过算子IR构图方式调用MatrixSetDiagV2算子。