• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1# Softmax
2
3
4## 概述
5
6给定一个tensor,计算其softmax结果。
7
8该算子对应的[NodeType](_n_n_rt.md#nodetype)为NODE_TYPE_SOFTMAX。
9
10输入:
11
12- x,n维tensor
13
14输出:
15
16- output,softmax的计算结果,一个n维tensor,类型和shape和x一致。
17
18**Since:**
19
203.2
21
22**Version:**
23
241.0
25
26**相关模块:**
27
28[NNRt](_n_n_rt.md)
29
30
31## 汇总
32
33
34### Public 属性
35
36  | 名称 | 描述 |
37| -------- | -------- |
38| [axis](#axis) | long[]<br/>指定计算softmax的维度。整数取值范围为[-n,n)。&nbsp; |
39
40
41## 类成员变量说明
42
43
44### axis
45
46
47```
48long [] Softmax::axis
49```
50**描述:**
51
52指定计算softmax的维度。整数取值范围为[-n,n)。
53