Ascend 950PR/Ascend 950DT
参数配置
输入
参数 |
维度 |
数据类型 |
格式 |
描述 |
|---|---|---|---|---|
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 |
矩阵乘计算结果。 |
父主题: 功能列表
