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. |