2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.mint.func_flatten.rst 958 Bytes
一键复制 编辑 原始数据 按行查看 历史
吴大维 提交于 2024-07-22 19:20 . add mint flatten docs

mindspore.mint.flatten

.. py:function:: mindspore.mint.flatten(input, start_dim=0, end_dim=-1)

    沿着从 `start_dim` 到 `end_dim` 的维度,对输入Tensor进行展平。

    参数:
        - **input** (Tensor) - 输入Tensor。
        - **start_dim** (int, 可选) - 要展平的第一个维度。默认值: ``0`` 。
        - **end_dim** (int, 可选) - 要展平的最后一个维度。默认值: ``-1`` 。

    返回:
        Tensor。如果没有维度被展平,返回原始的输入 `input`,否则返回展平后的Tensor。如果 `input` 是零维Tensor,将会返回一个一维Tensor。

    异常:
        - **TypeError** - `input` 不是Tensor。
        - **TypeError** - `start_dim` 或 `end_dim` 不是int类型。
        - **ValueError** - 规范化后, `start_dim` 大于 `end_dim` 。
        - **ValueError** - `start_dim` 或 `end_dim` 不在 [-input.dim, input.dim-1] 范围内。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助