Class ov::op::v8::Gather

class Gather : public ov::op::util::GatherBase

Gather slices from axis of data according to indices. Negative indices are supported and indicate reverse indexing from the end.

Public Functions

Gather(const Output<Node> &data, const Output<Node> &indices, const Output<Node> &axis, const int64_t batch_dims = 0)
Parameters

  • data – The tensor from which slices are gathered

  • indices – Tensor with indexes to gather

  • axis – The tensor is a dimension index to gather data from

  • batch_dims – The number of batch dimension in data and indices tensors.

virtual void validate_and_infer_types() override

Verifies that attributes and inputs are consistent and computes output shapes and element types. Must be implemented by concrete child classes so that it can be run any number of times.

Throws if the node is invalid.