昇腾社区首页
中文
注册
开发者
下载

asc_mmad_sparse

产品支持情况

[object Object]undefined

功能说明

完成矩阵乘加操作,传入的左矩阵A为稀疏矩阵,右矩阵B为稠密矩阵。对于矩阵A,在asc_mmad_sparse计算时完成稠密化;对于矩阵B,在计算执行前的输入数据准备时自行完成稠密化(按照下文中介绍的稠密算法进行稠密化),所以输入本接口的矩阵B为稠密矩阵。稠密矩阵B需要通过载入,同时加载索引矩阵,索引矩阵在矩阵B稠密化的过程中生成,再用于矩阵A的稠密化。

函数原型

  • 常规计算
[object Object]
  • 同步计算
[object Object]

参数说明

[object Object]undefined

返回值说明

流水类型

PIPE_MTE1

约束说明

  • 原始稀疏矩阵B每4个元素中应保证最多2个非零元素,如果存在3个或更多非零元素,则仅使用前2个非零元素。
  • 当m、k、n中的任意一个值为0时,该指令不会被执行。

稠密算法说明

假设原始稀疏矩阵B的每4个元素中至少有2个零,稠密化后的矩阵B是一个在每4个元素中过滤掉2个零的稠密矩阵。矩阵B稠密化的过程中生成索引矩阵,过程如下:对于稀疏矩阵B中的每4个元素,将在index矩阵中生成2个2位索引,并按照以下规则进行编码。索引必须在{0, 1, 2}范围内。 [object Object]• 第一个索引用于指示前3个元素中第1个非零元素的相对位置。 [object Object]• 第二个索引用于指示第2个非零元素在后3个元素中的相对位置。 [object Object]具体可参考下表。其中,“-” 表示算法不关心该位置上的值,因为其会被过滤。

[object Object][object Object][object Object][object Object][object Object][object Object][object Object]

调用示例

[object Object]