ngraph.opset1.ops.gather

ngraph.opset1.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], name: Optional[str] = None) _pyngraph.Node

Return Gather node which takes slices from axis of data according to indices.

Parameters
  • data – The tensor from which slices are gathered.

  • indices – Tensor with indexes to gather.

  • axis – The dimension index to gather data from.

  • name – Optional name for output node.

Returns

The new node performing a Gather operation on the data input tensor.