Gemm
功能
计算alpha乘以A与B的乘积,再与β和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,非0表示转置,0表示不转置。
- transB:数据类型支持bool,是否B需要转置,默认为0,非0表示转置,0表示不转置。
- alpha:数据类型支持float,该参数暂不支持。
- beta:数据类型支持float,该参数暂不支持。
输出
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算子清单