Method GatherND

GatherND(Tensor, TensorInt, Tensor, int)

Takes slices of values from the batched input tensor indexed by the indices tensor.

Declaration

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.