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