Gather
Description
Collect slices from "x" according to the corresponding axis.
Input
Two inputs:
x1: tensor of type float16 or float.
indices: tensor of type int32 or int64.
Output
One output:
y: tensor of the identical data type as input x1.
Attribute
axis: int, specifying the gather axis. Must be in the range [-r, r - 1], where r indicates the rank of the input data.
Constraints
Indexes with negative indices are not supported.
ONNX Opset Support
Opset v8/v9/v10/v11/v12/v13/v14/v15/v16/v17/v18
Parent topic: Operators Specifications of the AI Framework