mindspore2022/docs/api/api_python/nn/mindspore.nn.Flatten.rst

mindspore.nn.Flatten
====================

.. py:class:: mindspore.nn.Flatten

    对输入Tensor的第0维之外的维度进行展平操作。

    **输入：**

    - **x** (Tensor) - 要展平的输入Tensor。shape为 :math:`(N, *)`，其中 :math:`*` 表示任意的附加维度。数据类型为 `number <https://www.mindspore.cn/docs/api/zh-CN/master/api_python/mindspore.html#mindspore.dtype>`_。

    **输出：**

    Tensor，shape为 :math:`(N, X)`，其中 :math:`X` 是输入 `x` 的shape除N之外的其余维度的乘积。

    **异常：**

    **TypeError** - `x` 不是Tensor。

    **支持平台：**

    ``Ascend`` ``GPU`` ``CPU``

    **样例：**

    >>> x = Tensor(np.array([[[1.2, 1.2], [2.1, 2.1]], [[2.2, 2.2], [3.2, 3.2]]]), mindspore.float32)
    >>> net = nn.Flatten()
    >>> output = net(x)
    >>> print(output)
    [[1.2 1.2 2.1 2.1]
     [2.2 2.2 3.2 3.2]]
    >>> print(f"Before flatten the x shape is {x.shape}.")
    Before flatten the x shape is (2, 2, 2)
    >>> print(f"After flatten the output shape is {output.shape}.")
    After flatten the output shape is (2, 4)