2.6K Star 8.6K Fork 4.9K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
mindspore.mint.func_unbind.rst 943 Bytes
一键复制 编辑 原始数据 按行查看 历史
xiedejin1 提交于 2024-10-28 22:33 . add mint.unbind

mindspore.mint.unbind

.. py:function:: mindspore.mint.unbind(input, dim=0)

    根据指定轴对Tensor进行分解。给定一个shape为 :math:`(n_1, n_2, ..., n_R)` 的Tensor,在指定 `dim` 上对其分解,
    则返回多个shape为 :math:`(n_1, n_2, ..., n_{dim}, n_{dim+2}, ..., n_R)` 的Tensor。

    .. warning::
        这是一个实验性API,后续可能修改或删除。

    参数:
        - **input** (Tensor) - 用来分解的输入Tensor,其shape为 :math:`(n_1, n_2, ..., n_R)` 。其rank必须大于0。
        - **dim** (int,可选) - 用以分解的维度。取值范围为[-R, R)。默认值为 ``0`` 。

    返回:
        Tensor组成的tuple。每个Tensor对象的shape相同。

    异常:
        - **TypeError** - 如果 `input` 不是Tensor。
        - **TypeError** - 如果 `dim` 不是int类型。
        - **ValueError** - 如果 `dim` 超出[-R, R)范围。
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助

0d507c66 1850385 C8b1a773 1850385