Method GatherND
GatherND<T>(T, TensorInt, int)
Takes slices of values from the batched input tensor indexed by the indices
tensor.
Declaration
public T GatherND<T>(T X, TensorInt indices, int batchDims) where T : Tensor
Parameters
Type | Name | Description |
---|---|---|
T | X | The input tensor. |
TensorInt | indices | The indices tensor. |
int | batchDims | The number of batch dimensions of the input tensor, the gather begins at the next dimension. |
Returns
Type | Description |
---|---|
T | The computed output tensor. |
Type Parameters
Name | Description |
---|---|
T | The tensor type of the input and output tensors. |