API功能:完成量化的分组矩阵计算,然后所有组的矩阵计算结果相加后输出。
计算公式:
[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]):必选关键字参数,对应公式中的。数据类型支持[object Object],数据格式支持ND,shape支持2维,形状为(batch, m)。- 在实际计算时,
[object Object]会被广播到(batch,m,n)。
- 在实际计算时,
x2_scale (
[object Object]):必选关键字参数,对应公式中的。数据类型支持[object Object],数据格式支持ND,shape支持1维,形状为(n,)。- 在实际计算时,
[object Object]会被广播到(batch,m,n)。
- 在实际计算时,
[object Object]
公式中的,算子的计算结果。输出的数据类型为[object Object],数据格式为ND,shape为2维,形状为(m, n)。
- 该接口支持推理场景下使用。
- 该接口支持静态图模式。
- 传入的
[object Object]、[object Object]、[object Object]、[object Object]不能是空。 - 输入和输出支持以下数据类型组合:
[object Object]undefined
单算子调用
[object Object]图模式调用
[object Object]