openvino.runtime.opset6.gather

openvino.runtime.opset6.gather(data: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], indices: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], axis: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], name: Optional[str] = None) → openvino._pyopenvino.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.