Tensorflow.js tf.gatherND() Function
The tf.gatherND() function is used to gather slices from the shape specified by indices with the input tensor.
Parameters: This function accepts the following two parameters:
- tensor1: It is that tensor from which we gather values.
- tensor2: It is a K-dimensional integer tensor. It acts as indices. It must be of type int32.
Return Value: It returns tf.Tensor object.
Tensor [[2, 3], [3, 3]]
Tensor [[, ], [, ]]