OneHot

功能

根据输入生成独热编码张量

边界

【输入】

三个输入

indices:一个tensor,数据类型:int32、int64

depth:一个tensor,数据类型:int32、int64

values:一个tensor,数据类型:int32、int64、float16、float

【属性】

一个属性

axis:(可选)添加独热表示的轴

【输出】

一个输出

y:一个tensor,数据类型与values输入的类型一致

【约束】

算子属性不支持axis<-1

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

支持的ONNX版本

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