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]的张量。
此操作符假设每个输入特征都来自同一组类别。
如果输入是浮点数、int32或双精度张量,数据将被转换为整数,并使用cats_int64s类别列表进行查找。

属性

  • cats_int64s - 整型 :

    类别列表,整数。
    “cats_*”属性中必须且只能定义一个。

  • cats_strings - 字符串 :

    类别列表,字符串。
    “cats_*”属性中必须且只能定义一个。

  • zeros - 整型 (默认值为 '1')

    如果为true且类别不存在,将返回全零;如果为false且找不到类别,则操作符将失败。

输入

  • X (异构) - T

    要编码的数据。

输出

  • Y (异构) - tensor(float)

    编码后的输出数据,比X多一个维度。

类型约束

  • T 包含 ( tensor(double), tensor(float), tensor(int32), tensor(int64), tensor(string) )

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