forked from mindspore-Ecosystem/mindspore
33 lines
943 B
ReStructuredText
33 lines
943 B
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` 相同。
|
|||
|
|
|
|||
|
|
**异常：**
|
|||
|
|
|
|||
|
|
- **TypeError** - `block_size` 不是int类型。
|
|||
|
|
- **ValueError** - `block_size` 小于2。
|
|||
|
|
- **ValueError** - `x` 的维度不为4。
|