ngraph.opset5.gather_nd

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