Class ov::op::v15::Squeeze#

class Squeeze : public ov::op::util::SqueezeBase#

Squeeze operation.

Public Functions

Squeeze(const Output<Node> &data, const bool allow_axis_skip = false)#

Constructs a squeeze v15 operation.

Parameters:
  • dataInput tensor with data

  • allow_axis_skipShape inference result dynamic rank if selected axis has 1 in range of its dynamic

Squeeze(const Output<Node> &data, const Output<Node> &axes, const bool allow_axis_skip = false)#

Constructs a squeeze v15 operation.

Parameters:
  • dataInput tensor with data

  • axis – The axis along which to squeeze the input tensor.

  • allow_axis_skipShape inference result dynamic rank if selected axis has 1 in range of its dynamic

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.