开发者
资源

Ascend 950PR/Ascend 950DT

参数配置

参数

取值范围

特殊说明

transposeA

false/true

取值为true时,部分场景不支持,详见规格说明

transposeB

false/true

-

hasBias

false/true

取值为true时,部分场景不支持,详见规格说明

outDataType

ACL_DT_UNDEFINED

-

enAccum

false

-

matmulType

MATMUL_UNDEFINED

-

quantMode

QUANT_UNDEFINED

-

输入

参数

维度

数据类型

格式

描述

x

[m, k]/[batch, m, k]

float16/bf16

ND

矩阵乘的A矩阵。

weight

[k, n]/[batch, k, n]

float16/bf16

ND

矩阵乘的B矩阵,权重。

bias

[1, n]/[n]

float16/bf16/float

ND

叠加的偏置矩阵。当hasBias取值为true时输入。

输出

参数

维度

数据类型

格式

描述

output

[m, n]/[batch, m, n]

float16/bf16

ND

矩阵乘计算结果。

规格说明

图1 输入输出属性排列组合
  • transposeA取值为true时,不支持组合2、5、8、11
  • 组合3、6、9、12不支持叠加偏置。