Method GatherElements
GatherElements(Tensor, TensorInt, Tensor, int)
Takes values from the input tensor indexed by the indices tensor along a given axis and concatenates them.
Declaration
void GatherElements(Tensor X, TensorInt indices, Tensor O, int axis)Parameters
| Type | Name | Description |
|---|---|---|
| Tensor | X | The input tensor. |
| TensorInt | indices | The indices tensor. |
| Tensor | O | The output tensor to be computed and filled. |
| int | axis | The axis along which to gather. |