squeeze_cn.rst

.. _cn_api_paddle_tensor_squeeze
squeeze
-------------------------------

.. py:function:: paddle.tensor.squeeze(input, zxes, name=None)

该OP会根据axes压缩输入Tensor的维度。如果指定了axes，则会删除axes中指定的维度，axes指定的维度要等于1。如果没有指定axes，那么所有等于1的维度都会被删除。

**参数**：
        - **input** (Variable) - 输入任意维度的Tensor。 支持的数据类型：float32，float64，int8，int32，int64。
        - **axes** (list) - 输入一个或一列整数，代表要压缩的轴。axes的范围： [−rank(input),rank(input))] 。 axes为负数时， axes=axes+rank(input) 。
        - **name** (str，可选) - 一般无需设置，默认值为None。

**返回**：返回对维度进行压缩后的Tensor。数据类型与输入Tensor一致。

**返回类型**：Variable

**代码示例**：

.. code-block:: python

    import paddle
    import paddle.fluid as fluid
    import numpy as np
    with fluid.dygraph.guard():
        input_1 = np.random.random([5, 1, 10]).astype("int32")
        # input is a variable which shape is [5, 1, 10]
        input = fluid.dygraph.to_variable(input_1)

        output = paddle.fluid.layers.squeeze(input, axes=[1])
        # output.shape [5, 10]