2.3K Star 8.1K Fork 4.3K

GVPMindSpore / mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.func_unsqueeze.rst 843 Bytes
一键复制 编辑 原始数据 按行查看 历史

mindspore.ops.unsqueeze

.. py:function:: mindspore.ops.unsqueeze(input, dim)

    对输入 `input` 在给定维上添加额外维度。

    扩展后的Tensor中位置对应 `dim` 的维度为插入的新维度。

    参数:
        - **input** (Tensor) - 输入Tensor,shape为 :math:`(n_1, n_2, ..., n_R)`。
        - **dim** (int) - 新插入的维度的位置。 `dim` 的值必须在范围 `[-input.ndim-1, input.ndim]` 内。仅接受常量输入。

    返回:
        Tensor,维度在指定轴扩展之后的Tensor,与 `input` 的数据类型相同。如果 `dim` 是0,那么它的shape为 :math:`(1, n_1, n_2, ..., n_R)`。

    异常:
        - **TypeError** - 如果 `dim` 不是int。
        - **ValueError** - 如果 `dim` 超出了 :math:`[-input.ndim-1, input.ndim]` 的范围。
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
master

搜索帮助