OneHot

功能

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

输入

• indices：输入Tensor，数据类型支持int32
• depth：输入Tensor，数据类型支持int32
• values：输入Tensor，数据类型支持int32、float16、float。

属性

axis（可选）：数据类型为int64，添加算子表示的轴。

输出

y：输出Tensor，数据类型与输入indices一致。

约束与限制

• 算子属性不支持axis<-1。
• depth >= 1
• 当输出为int64时，最后一个输入值values=[0,1]，1代表on_value，0代表off_value。

支持的ONNX版本

Opset v9/v10/v11/v12/v13/v14/v15/v16/v17/v18。