openvino.runtime.opset6.gather_elements

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