forked from mindspore-Ecosystem/mindspore
29 lines
1.0 KiB
ReStructuredText
29 lines
1.0 KiB
ReStructuredText
mindspore.ops.SpaceToDepth
|
||
==========================
|
||
|
||
.. py:class:: mindspore.ops.SpaceToDepth(block_size)
|
||
|
||
将空间维度分块，增加Tensor深度。
|
||
|
||
输出Tensor的高度为（输入高度 / `block_size`）。
|
||
|
||
输出Tensor的宽度为（输入宽度 / `block_size`）。
|
||
|
||
输出Tensor的深度为（输入深度 * `block_size` * `block_size`）。
|
||
|
||
输入Tensor的高度和宽度必须可被 `block_size` 整除。格式为"NCHW"（batch_size，深度，高度，宽度）。
|
||
|
||
参数：
|
||
- **block_size** (int) - 用于划分空间维度的子块的大小。必须>=2。
|
||
|
||
输入：
|
||
- **x** (Tensor) - 4维Tensor。数据类型为Number。
|
||
|
||
输出：
|
||
4维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。
|