ai.onnx.ml - OneHotEncoder

OneHotEncoder - 1 (ai.onnx.ml)

版本

  • 名称: OneHotEncoder (GitHub)

  • : ai.onnx.ml

  • 起始版本: 1

  • 函数: False

  • 支持级别: SupportType.COMMON

  • 形状推断: True

该算子的此版本**自 ai.onnx.ml 域的 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 in ( tensor(double), tensor(float), tensor(int32), tensor(int64), tensor(string) )

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