Class ov::op::v0::Squeeze#

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

Squeeze operation.

Public Functions

Squeeze(const Output<Node> &data)#

Constructs a squeeze v0 operation.

Parameters:

dataInput tensor with data

Squeeze(const Output<Node> &data, const Output<Node> &axes)#

Constructs a squeeze v0 operation.

Parameters:
  • dataInput tensor with data

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

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.