Method GatherND
GatherND(Tensor, TensorInt, Tensor, int)
Takes slices of values from the batched input tensor indexed by the indices tensor.
Declaration
public override void GatherND(Tensor X, TensorInt indices, Tensor O, int batchDims)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 | batchDims | The number of batch dimensions of the input tensor, the gather begins at the next dimension. |