ngraph.gather¶
- ngraph.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 with support of negative indices.
- Parameters
data – N-D tensor with data for gathering
indices – N-D tensor with indices by which data is gathered. Negative indices
indicate reverse indexing from the end :param axis: axis along which elements are gathered :param batch_dims: number of batch dimensions :return: The new node which performs Gather