ai.onnx.ml - ArrayFeatureExtractor¶

ArrayFeatureExtractor - 1 (ai.onnx.ml)¶

版本¶

名稱: ArrayFeatureExtractor (GitHub)
網域: ai.onnx.ml
since_version: 1
函式: False
支援等級: SupportType.COMMON
形狀推斷: True

此版本的運算子自網域 ai.onnx.ml 的版本 1 開始提供。

摘要¶

根據傳遞的索引選取輸入張量的元素。
這些索引會套用至張量的最後一個軸。

輸入¶

X (異質) - T

要選取的資料
Y (異質) - tensor(int64)

索引，以 0 作為任何維度的第一個索引。

輸出¶

Z (異質) - T

選取的輸出資料為陣列

類型限制¶

T 在 ( 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",
)