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