ai.onnx.ml - OneHotEncoder¶
OneHotEncoder - 1 (ai.onnx.ml)¶
版本¶
域:
ai.onnx.ml
起始版本:
1
函数:
False
支持级别:
SupportType.COMMON
形状推断:
True
此版本的算子自域 ai.onnx.ml 的版本 1 起可用。
摘要¶
将每个输入元素替换为一个由 1 和 0 组成的数组,其中一个 1 放置在输入类别的索引处。总类别计数将决定输出数组 Y 的额外维度的尺寸。
例如,如果输入一个值为 4 的张量,且类别计数为 8,则输出将是一个张量,其值为 [0,0,0,0,1,0,0,0]
。
此算子假定每个输入特征来自同一组类别。
如果输入是 float、int32 或 double 类型的张量,则数据将被转换为整数,并使用 cats_int64s 类别列表进行查找。
属性¶
cats_int64s - INTS :
类别列表,整数。
必须且只能定义一个 'cats_*' 属性。cats_strings - STRINGS :
类别列表,字符串。
必须且只能定义一个 'cats_*' 属性。zeros - INT (默认值为
'1'
)如果为 true 且类别不存在,将返回全零;如果为 false 且找不到类别,算子将失败。
输入¶
X (异构) - T
待编码的数据。
输出¶
Y (异构) - tensor(float)
编码后的输出数据,比 X 多一个维度。
类型约束¶
T 属于 (
tensor(double)
,tensor(float)
,tensor(int32)
,tensor(int64)
,tensor(string)
)输入必须是数值类型的张量。