开发者
资源
[object Object][object Object][object Object]undefined
[object Object]
  • API功能:完成量化的分组矩阵计算,然后所有组的矩阵计算结果相加后输出。

  • 计算公式:

out=i=0batch(x1i@x2i)x1Scaleix2Scaleout = \sum_{i=0}^{batch}(x1_i @ x2_i) * x1Scale_i * x2Scale [object Object]
[object Object]
[object Object]
  • x1 ([object Object]):必选参数,数据类型支持[object Object],数据格式支持ND,shape支持3维,形状为(batch, m, k)。

  • x2 ([object Object]):必选参数,数据类型支持[object Object],数据格式支持NZ,shape支持3维,形状为(batch, k, n)。

    • 可通过[object Object]将ND格式的x2转换为NZ格式。
  • x1_scale ([object Object]):必选关键字参数,对应公式中的x1Scalex1Scale。数据类型支持[object Object],数据格式支持ND,shape支持2维,形状为(batch, m)。

    • 在实际计算时,[object Object]会被广播到(batch,m,n)。
  • x2_scale ([object Object]):必选关键字参数,对应公式中的x2Scalex2Scale。数据类型支持[object Object],数据格式支持ND,shape支持1维,形状为(n,)。

    • 在实际计算时,[object Object]会被广播到(batch,m,n)。
[object Object]

[object Object]

公式中的outout,算子的计算结果。输出的数据类型为[object Object],数据格式为ND,shape为2维,形状为(m, n)。

[object Object]
  • 该接口支持推理场景下使用。
  • 该接口支持静态图模式。
  • 传入的[object Object][object Object][object Object][object Object]不能是空。
  • 输入和输出支持以下数据类型组合:[object Object]undefined
[object Object]
  • 单算子调用

    [object Object]
  • 图模式调用

    [object Object]