MatMulFusion
概述
对输入x1和x2,计算x1和x2的内积。
该算子对应的NodeType为NODE_TYPE_MATMUL_FUSION。
输入:
-
x1,n维输入tensor,实数或DATA_TYPE_BOOL类型的tensor。
-
x2,n维输入tensor,实数或DATA_TYPE_BOOL类型的tensor。
输出:
- output,计算得到内积,当type!=DATA_TYPE_UNKNOWN时,output数据类型由type决定;当type==DATA_TYPE_UNKNOWN时, output的数据类型取决于x1和x2进行计算时转化的数据类型。
Since:
3.2
Version:
1.0
相关模块:
汇总
Public 属性
名称 | 描述 |
---|---|
transposeA | boolean 是否对x1矩阵进行转置。 |
transposeB | boolean 是否对x2矩阵进行转置。 |
activationType | enum ActivationType 激活函数,详情请参考:ActivationType。 |
类成员变量说明
activationType
enum ActivationType MatMulFusion::activationType
描述: 激活函数,详情请参考:ActivationType。
transposeA
boolean MatMulFusion::transposeA
描述: 是否对x1矩阵进行转置。
transposeB
boolean MatMulFusion::transposeB
描述: 是否对x2矩阵进行转置。