harmony 鸿蒙ArgMaxFusion

  • 2023-06-24
  • 浏览 (646)

ArgMaxFusion

概述

返回跨轴的tensor前K个索引或者是数值。

该算子对应的NodeType为NODE_TYPE_ARGMAX_FUSION。

输入:

  • x,n维tensor,输入张量(N,*),其中*意味着数量任意的附加维度。

输出:

  • output,轴上输入张量最大值的前K个索引或者是数值。

Since: 3.2 Version: 1.0 相关模块:

NNRt

汇总

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 鸿蒙HDI 参数参考

harmony 鸿蒙ActRecognitionEvent

harmony 鸿蒙Activation

harmony 鸿蒙ActivityRecognitionTypes.idl

harmony 鸿蒙AddFusion

harmony 鸿蒙Alignment

harmony 鸿蒙AllocInfo

harmony 鸿蒙ApduSimIORequestInfo

harmony 鸿蒙Audio

harmony 鸿蒙AudioAdapterDescriptor

0  赞