harmony 鸿蒙Softmax
Softmax
概述
给定一个tensor，计算其softmax结果。
该算子对应的NodeType为NODE_TYPE_SOFTMAX。
输入：
- x，n维tensor
输出：
- output，softmax的计算结果，一个n维tensor，类型和shape和x一致。
Since:
3.2
Version:
1.0
相关模块:
汇总
Public 属性
| 名称 | 描述 |
|---|---|
| axis | long[] 指定计算softmax的维度。整数取值范围为[-n，n)。 |
类成员变量说明
axis
long [] Softmax::axis
描述:
指定计算softmax的维度。整数取值范围为[-n，n)。
你可能感兴趣的鸿蒙文章
harmony 鸿蒙ActivityRecognitionTypes.idl
0
赞
- 所属分类： 后端技术
- 本文标签：
热门推荐
-
2、 - 优质文章
-
3、 gate.io
-
8、 golang
-
9、 openharmony
-
10、 Vue中input框自动聚焦