Gemm
功能
计算alpha乘以A与B的乘积,再与beta和input C的乘积求和。

输入
- A:输入二维矩阵Tensor,数据类型支持float16、float,数据类型需要与C、B构成互相推导关系,shape(或者转置后shape)需要满足与B相乘条件,数据格式支持ND。
- B:输入二维矩阵Tensor,数据类型支持float16、float,数据类型需要与C,A构成互相推导关系,shape(或者转置后shape)需要满足与A相乘条件,数据格式支持ND。
- C(可选):偏差。
属性
- transA:数据类型支持bool,表示A是否需要转置,默认为0,1表示转置,0表示不转置。
- transB:数据类型支持bool,B表示是否需要转置,默认为0,1表示转置,0表示不转置。
- alpha:数据类型支持float,输入张量 A * B 乘积的标量乘数。
- beta:数据类型支持float,输入张量 C 的标量乘数。
输出
Y:输出二维矩阵Tensor,shape需要与A@B计算后的结果一致,数据类型需要与C构成互相推导关系,数据类型支持float16、float。
约束
v8/v9/v10版本不支持atc工具参数--precision_mode=must_keep_origin_dtype时float类型输入。
支持的ONNX版本
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18
父主题: 支持ONNX算子清单