ngraph.opset7.ops.gather

ngraph.opset7.ops.gather(data: Union[_pyngraph.Node, int, float, numpy.ndarray], indices: Union[_pyngraph.Node, int, float, numpy.ndarray], axis: Union[_pyngraph.Node, int, float, numpy.ndarray], batch_dims: Optional[int] = 0) _pyngraph.Node

Return a node which performs Gather.

Parameters
  • data – N-D tensor with data for gathering

  • indices – N-D tensor with indices by which data is gathered

  • axis – axis along which elements are gathered

  • batch_dims – number of batch dimensions

:return:: The new node which performs Gather