Method GatherND
GatherND(Tensor, TensorInt, Tensor, int)
Takes slices of values from the batched input tensor indexed by the indices
tensor.
Declaration
public virtual 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. |