1# OneHot
2
3
4## 概述
5
6根据indices指定的位置,生成一个由one-hot向量构成的tensor。
7
8每个onehot向量中的有效值由on_value决定,其他位置由off_value决定。
9
10该算子对应的[NodeType](_n_n_rt_v20.md#nodetype)为NODE_TYPE_ONE_HOT。
11
12输入:
13
14- indices,n维tensor。indices中每个元素决定每个one-hot向量,on_value的位置。
15
16- depth,一个整型标量,决定one-hot向量的深度。要求depth>0。
17
18- on_value,一个标量,指定one-hot向量中的有效值。
19
20- off_value,一个标量,指定one-hot向量中除有效位以外,其他位置的值。
21
22输出:
23
24- output,如果indices时n维tensor,则output是(n+1)维tensor。output的形状由indices和axis共同决定。
25
26**起始版本:** 3.2
27
28**相关模块:**[NNRt](_n_n_rt_v20.md)
29
30
31## 汇总
32
33
34### Public 属性
35
36| 名称 | 描述 |
37| -------- | -------- |
38| long [axis](#axis) | 一个整型标量,指定插入one-hot的维度。 indices的形状是[N,C],depth的值是D,当axis=0时,output形状为[D,N,C], indices的形状是[N,C],depth的值是D,当axis=-1时,output形状为[N,C,D], indices的形状是[N,C],depth的值是D,当axis=1时,output形状为[N,D,C]。  |
39
40
41## 类成员变量说明
42
43
44### axis
45
46```
47long OneHot::axis
48```
49**描述**
50
51一个整型标量,指定插入one-hot的维度。 indices的形状是[N,C],depth的值是D,当axis=0时,output形状为[D,N,C], indices的形状是[N,C],depth的值是D,当axis=-1时,output形状为[N,C,D], indices的形状是[N,C],depth的值是D,当axis=1时,output形状为[N,D,C]。
52