ai.onnx.ml - OneHotEncoder

OneHotEncoder - 1 (ai.onnx.ml)

版本

  • 名称: OneHotEncoder (GitHub)

  • : 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) )

    输入必须是数值类型的张量。