harmony 鸿蒙MatMulFusion
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进行计算时转化的数据类型。
起始版本: 3.2
相关模块:NNRt
汇总
Public 属性
名称 | 描述 |
---|---|
boolean transposeA | 是否对x1矩阵进行转置。 |
boolean transposeB | 是否对x2矩阵进行转置。 |
enum ActivationTypeactivationType | 激活函数,详情请参考:ActivationType。 |
类成员变量说明
activationType
enum ActivationType MatMulFusion::activationType
描述
激活函数,详情请参考:ActivationType。
transposeA
boolean MatMulFusion::transposeA
描述
是否对x1矩阵进行转置。
transposeB
boolean MatMulFusion::transposeB
描述
是否对x2矩阵进行转置。
你可能感兴趣的鸿蒙文章
0
赞
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
8、 golang
-
9、 openharmony
-
10、 Vue中input框自动聚焦