• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Softmax
2
3## 概述
4
5给定一个tensor,计算其softmax结果。
6
7该算子对应的[NodeType](_n_n_rt_v20.md#nodetype)为NODE_TYPE_SOFTMAX。
8
9输入:
10
11- x,n维tensor。
12
13输出:
14
15- output,softmax的计算结果,一个n维tensor,类型和shape和x一致。
16
17**起始版本:** 3.2
18
19**相关模块:**[NNRt](_n_n_rt_v20.md)
20
21## 汇总
22
23### Public 属性
24
25| 名称 | 描述 |
26| -------- | -------- |
27| long[] [axis](#axis) | 指定计算softmax的维度。整数取值范围为[-n,n)。  |
28
29## 类成员变量说明
30
31### axis
32
33```
34long [] Softmax::axis
35```
36**描述**
37
38指定计算softmax的维度。整数取值范围为[-n,n)。
39