OneHot

功能

根据输入生成独热编码tensor。

输入

三个输入：

indices：一个tensor，数据类型：int32、int64。

depth：一个tensor，数据类型：int32、int64。

values：一个tensor，数据类型：int32、int64、float16、float。

输出

一个输出：

y：一个tensor，数据类型与values输入的类型一致。

属性

axis：（可选）添加独热表示的轴。

约束

算子属性不支持axis<-1。

当输出为int64时，最后一个输入值values=[0,1]，1代表on_value，0代表off_value。

支持的ONNX版本

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