Squeeze

Removes all dimensions that have length 1.

Any input data type and shape is supported. The output shape will be the same as the input, but with all dimensions with size 1 removed. The output data type will be the same as the input.

For example, given an input of shape (A, B, 1, C), the output shape will be (A, B, C).

Dimensions that have an unkown size will not be removed, even if they have the size 1 when actually running the model.