GatherElements

Description

Obtains the index generated by the input tensor indices, determines the element of the input tensor input, and generates the output tensor output at the position corresponding to the input tensor indices.

Input

• input: input tensor of type float16, float, or int32. The rank is greater than or equal to 1.
• indices: input tensor of type int32 or int64. The rank is the same as that of input.

Attribute

axis: int, indicating the axis for aggregation. The default value is 0. A negative value indicates that the dimension is calculated from the end.

Output

output: output tensor with the same shape as indices.

Constraints

None.

ONNX Opset Support

Opset v11/v12/v13/v14/v15/v16/v17/v18