squeeze

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

    • 例1：
    • 例3：

    参数：

    • axes (list) - 输入一个或一列整数，代表要压缩的轴。axes的范围：

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

    代码示例：