ngraph.gather_nd¶
-
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.
- Parameters
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
- Return:
The new node which performs GatherND