ai.onnx.ml - 二值化器

二值化器 - 1 (ai.onnx.ml)

版本

• 名称: Binarizer (GitHub)

• 领域: ai.onnx.ml

• since_version: 1

• 函数: False

• support_level: SupportType.COMMON

• 形状推断: True

此版本的运算符自ai.onnx.ml域的第1版起已可用。

摘要

根据与阈值比较的结果，将输入张量的值逐元素映射为0或1。

属性

• threshold - FLOAT (默认为 '0.0'):

  大于此值的值映射为1，其他值映射为0。

输入

• X (异构) - T:

  要二值化的数据

输出

• Y (异构) - T:

  二值化输出数据

类型约束

• T 在 ( tensor(double), tensor(float), tensor(int32), tensor(int64) ) 中:

  输入必须是一个数值类型的张量。输出将是相同类型的张量。

示例

默认

import numpy as np
import onnx

threshold = 1.0
node = onnx.helper.make_node(
    "Binarizer",
    inputs=["X"],
    outputs=["Y"],
    threshold=threshold,
    domain="ai.onnx.ml",
)
x = np.random.randn(3, 4, 5).astype(np.float32)
y = compute_binarizer(x, threshold)[0]

expect(node, inputs=[x], outputs=[y], name="test_ai_onnx_ml_binarizer")