功能
对长度为n的输入Tensor, 经过one_hot的计算后得到一个元素数量为n*k的输出Tensor,其中k的值为numClasses。

输入
- indices:输入Tensor,数据类型支持int32、int64。
- depth:输入Tensor,数据类型支持int32、int64。
- values:输入Tensor,数据类型支持int32、int64、float16、float。
属性
axis(可选):数据类型为int64,添加算子表示的轴。
输出
y:输出Tensor,数据类型与输入indices一致。
约束与限制
- 算子属性不支持axis<-1。
- 当输出为int64时,最后一个输入值values=[0,1],1代表on_value,0代表off_value。
支持的ONNX版本
Opset v9/v10/v11/v12/v13/v14/v15/v16/v17/v18。