mindspore/docs/api/api_python/ops/mindspore.ops.Flatten.rst

mindspore.ops.Flatten
======================

.. py:class:: mindspore.ops.Flatten

    扁平化（Flatten）输入Tensor，不改变0轴的size。

    **输入：**

    - **input_x** (Tensor) - 待扁平化的Tensor，其shape为 :math:`(N, \ldots)`， :math:`N` 表示batch size。

    **输出：**

    Tensor，输出shape为 :math:`(N, X)` 的Tensor，其中 :math:`X` 是余下维度的乘积。

    **异常：**

    - **TypeError** - `input_x` 不是Tensor。
    - **ValueError** - `input_x` 的shape长度小于1。