2.6K Star 8.6K Fork 4.8K

GVPMindSpore/mindspore

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
克隆/下载
mindspore.ops.SpaceToDepth.rst 1.02 KB
一键复制 编辑 原始数据 按行查看 历史
俞涵 提交于 2023-03-16 17:29 . modify math format error

mindspore.ops.SpaceToDepth

.. py:class:: mindspore.ops.SpaceToDepth(block_size)

    将空间维度分块,增加Tensor深度。

    输出Tensor的高度为 :math:`height / block\_size`。

    输出Tensor的宽度为 :math:`weight / block\_size`。

    输出Tensor的深度为 :math:`block\_size * block\_size * input\_depth`。

    输入Tensor的高度和宽度必须可被 `block_size` 整除。格式为"NCHW"(batch_size,深度,高度,宽度)。

    参数:
        - **block_size** (int) - 用于划分空间维度的子块的大小。必须>=2。

    输入:
        - **x** (Tensor) - 四维Tensor。数据类型为Number。

    输出:
        四维Tensor,数据类型与 `x` 相同,shape: :math:`(N, (C_{in} * \text{block_size} * 2), H_{in} / \text{block_size}, W_{in} / \text{block_size})` 。

    异常:
        - **TypeError** - `block_size` 不是int类型。
        - **ValueError** - `block_size` 小于2。
        - **ValueError** - `x` 的维度不为4。
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Python
1
https://gitee.com/mindspore/mindspore.git
git@gitee.com:mindspore/mindspore.git
mindspore
mindspore
mindspore
r2.3.1

搜索帮助