- ngraph.gather_nd(data: Union[_pyngraph.Node, int, float, numpy.ndarray], indices: Union[_pyngraph.Node, int, float, numpy.ndarray], batch_dims: Optional[int] = 0, name: Optional[str] = None) _pyngraph.Node ¶
Return a node which performs GatherND.
data – N-D tensor with data for gathering
indices – K-D tensor of tuples with indices by which data is gathered
batch_dims – Scalar value of batch dimensions
The new node which performs GatherND