Gemm
功能
通用矩阵乘。
输入
三个输入:
A:2D矩阵tensor,数据类型:float16、float。
B:2D矩阵tensor,数据类型:float16、float。
C:偏差,可选,该参数暂不支持。
输出
一个输出:
Y:2D矩阵tensor,数据类型:float16、float。
属性
transA:bool,是否A需要转置。
transB:bool,是否B需要转置。
alpha:float,该参数暂不支持。
beta: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算子清单