OneHot
功能
根据输入生成独热编码张量
边界
【输入】
三个输入
indices:一个tensor,数据类型:uint8,uint16, uint32,uint64,int8,int16,int32,int64,float16,float,double
depth:一个tensor,数据类型:uint8,uint16, uint32,uint64,int8,int16,int32,int64,float16,float,double
values:一个tensor,数据类型:uint8,uint16, uint32,uint64,int8,int16,int32,int64,float16,float,double
【属性】
一个属性
axis:(可选)添加独热表示的轴
【输出】
一个输出
y:一个tensor,数据类型与values输入的类型一致
【约束】
算子属性不支持axis<-1
支持的ONNX版本
Opset v9/v10/v11/v12/v13
父主题: 支持ONNX算子清单