ai.onnx.ml - 数组特征提取器

数组特征提取器 - 1 (ai.onnx.ml)

版本

• 名称: 数组特征提取器 (GitHub)

• 域: ai.onnx.ml

• 自版本: 1

• 函数: False

• 支持级别: SupportType.COMMON

• 形状推断: True

此版本的运算符已在域 ai.onnx.ml 的版本 1 中可用。

摘要

根据传递的索引选择输入张量的元素。
索引应用于张量的最后一个轴。

输入

• X (异构) - T

  要选择的數據

• Y (异构) - tensor(int64)

  索引，基于 0 作为任何维度的第一个索引。

输出

• Z (异构) - T

  作为数组选择的输出数据

类型约束

• T in ( tensor(double), tensor(float), tensor(int32), tensor(int64), tensor(string) )

  输入必须是数字类型或字符串的张量。输出将具有相同的张量类型。

示例

默认值

import numpy as np
import onnx

node = onnx.helper.make_node(
    "ArrayFeatureExtractor",
    inputs=["x", "y"],
    outputs=["z"],
    domain="ai.onnx.ml",
)

x = np.arange(12).reshape((3, 4)).astype(np.float32)
y = np.array([0, 1], dtype=np.int64)
z = np.array([[0, 4, 8], [1, 5, 9]], dtype=np.float32).T
expect(
    node,
    inputs=[x, y],
    outputs=[z],
    name="test_ai_onnx_ml_array_feature_extractor",
)