Class ov::op::v1::Split

class Split : public ov::op::Op

Splits the input tensor into a list of equal sized tensors.

Public Functions

Split() = default

Constructs a split operation.

Split(const Output<Node> &data, const Output<Node> &axis, const size_t num_splits)

Constructs a split operation.

Parameters
  • data – The tensor to be split.

  • axis – The index of an axis in “data” along which to perform the split.

  • num_splits – The number of pieces that the data tensor should be split into.

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.

virtual bool has_evaluate() const override

Allows to get information about availability of evaluate method for the current operation.