Versioned name: Squeeze-1
Category: Shape manipulation
Short description: Squeeze removes specified dimensions (second input) equal to 1 of the first input tensor. If the second input is omitted then all dimensions equal to 1 are removed. If the specified dimension is not equal to one then error is raised.
Attributes: Squeeze operation doesn't have attributes.
Inputs:
- 1: Multidimensional input tensor of type T. Required.
- 2: 0D or 1D tensor of type T_SHAPE with dimensions indices to squeeze. Values could be negative. Optional.
Types
- T: supported type.
- T_SHAPE: supported integer type.
Example
Example 1:
<layer ... type="Squeeze">
<input>
<port id="0">
<dim>1</dim>
<dim>3</dim>
<dim>1</dim>
<dim>2</dim>
</port>
</input>
<input>
<port id="1">
<dim>2</dim>
</port>
</input>
<output>
<port id="2">
<dim>3</dim>
<dim>2</dim>
</port>
</output>
</layer>
Example 2: squeeze 1D tensor with 1 element to a 0D tensor (constant)
<layer ... type="Squeeze">
<input>
<port id="0">
<dim>1</dim>
</port>
</input>
<input>
<port id="1">
<dim>1</dim>
</port>
</input>
<output>
<port id="2">
</port>
</output>
</layer>