harmony 鸿蒙ArgMaxFusion
ArgMaxFusion
概述
返回跨轴的tensor前K个索引或者是数值。
该算子对应的NodeType为NODE_TYPE_ARGMAX_FUSION。
输入:
- x,n维tensor,输入张量(N,*),其中*意味着数量任意的附加维度。
输出:
- output,轴上输入张量最大值的前K个索引或者是数值。
Since: 3.2 Version: 1.0 相关模块:
汇总
Public 属性
名称 | 描述 |
---|---|
axis | long 指定求最大值索引的维度。 |
topK | long 轴上前K个最大值。 |
keepDims | boolean 是否保持输出维度和输入的维度是否相同。 |
outMaxValue | boolean 若为false则输出索引,为true则输出数值,默认为false。 |
类成员变量说明
axis
long ArgMaxFusion::axis
描述: 指定求最大值索引的维度。
keepDims
boolean ArgMaxFusion::keepDims
描述: 是否保持输出维度和输入的维度是否相同。
outMaxValue
boolean ArgMaxFusion::outMaxValue
描述: 若为false则输出索引,为true则输出数值,默认为false。
topK
long ArgMaxFusion::topK
描述: 轴上前K个最大值。
你可能感兴趣的鸿蒙文章
harmony 鸿蒙ActivityRecognitionTypes.idl
0
赞
- 所属分类: 后端技术
- 本文标签:
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
8、 golang
-
9、 openharmony
-
10、 Vue中input框自动聚焦