昇腾社区首页
中文
注册

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