ngraph.gather_elements¶
- ngraph.gather_elements(data: Union[_pyngraph.Node, int, float, numpy.ndarray], indices: Union[_pyngraph.Node, int, float, numpy.ndarray], axis: Optional[int] = 0, name: Optional[str] = None) _pyngraph.Node¶
Return a node which performs GatherElements.
- 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
- Returns
The new node which performs GatherElements