mindspore.nn.Unflatten
=======================

.. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/r2.2/resource/_static/logo_source.svg
    :target: https://gitee.com/mindspore/mindspore/blob/r2.2/docs/api/api_python/nn/mindspore.nn.Unflatten.rst
    :alt: 查看源文件


.. py:class:: mindspore.nn.Unflatten(axis, unflattened_size)

    根据 `axis` 和 `unflattened_size` 折叠指定维度为给定形状。
    
    参数：
        - **axis** (int) - 指定输入Tensor被折叠维度。
        - **unflattened_size** (Union(tuple[int], list[int])) - 指定维度维度折叠后的新shape，可以为tuple[int]或者list[int]。 `unflattened_size` 中各元素的乘积必须等于input_shape[axis]。

    输入：
        - **input** (Tensor) - 进行折叠操作的Tensor。

    输出：
        折叠操作后的Tensor。

    异常：
        - **TypeError** - `axis` 不是int。
        - **TypeError** - `unflattened_size` 既不是tuple[int]也不是list[int]。
        - **TypeError** - `unflattened_size` 中各元素的乘积不等于input_shape[axis]。