开发者
下载

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