Softmax
概述
给定一个tensor，计算其softmax结果。
该算子对应的NodeType为NODE_TYPE_SOFTMAX。
输入：
输出：
- output，softmax的计算结果，一个n维tensor，类型和shape和x一致。
起始版本： 3.2
相关模块：NNRt
汇总
Public 属性
| 名称 |
描述 |
| long[] axis |
指定计算softmax的维度。整数取值范围为[-n，n)。 |
类成员变量说明
axis
long [] Softmax::axis
描述
指定计算softmax的维度。整数取值范围为[-n，n)。
你可能感兴趣的鸿蒙文章
harmony 鸿蒙NNRt HDI 参考
harmony 鸿蒙Activation
harmony 鸿蒙Activation
harmony 鸿蒙AddFusion
harmony 鸿蒙AddFusion
harmony 鸿蒙ArgMaxFusion
harmony 鸿蒙ArgMaxFusion
harmony 鸿蒙AvgPoolFusion
harmony 鸿蒙AvgPoolFusion
harmony 鸿蒙BatchToSpaceND