mindspore/docs/api/api_python/ops/mindspore.ops.func_col2im.rst

25 lines
1.5 KiB
ReStructuredText
Raw Permalink Normal View History

mindspore.ops.col2im
====================
.. py:function:: mindspore.ops.col2im(input_x, output_size, kernel_size, dilation, padding_value, stride)
2022-12-29 16:33:21 +08:00
将一组滑动局部块组合成一个大的Tensor。
2022-07-25 15:09:45 +08:00
参数:
- **input_x** (Tensor) - 四维Tensor输入的批量的滑动局部块数据类型支持float16和float32。
2022-12-29 10:52:15 +08:00
- **output_size** (Tensor) - 包含两个int元素的一维Tensor输出张量的后两维的shape。
2022-07-25 15:09:45 +08:00
- **kernel_size** (Union[int, tuple[int], list[int]]) - 滑动窗口的大小。tuple的两个元素分别对应kernel的高度与宽度。如果为一个int则kernel的高度与宽度均为该值。
- **dilation** (Union[int, tuple[int], list[int]]) - 滑动窗口扩张的大小。
- **padding_value** (Union[int, tuple[int], list[int]]) - 填充的大小。
- **stride** (Union[int, tuple[int], list[int]]) - 步长的大小。
返回:
Tensor输出的张量维度和类型和输入一致。
异常:
- **TypeError** - 如果 `kernel_size``dilation``padding_value``stride` 不属于 Union[int, tuple[int], list[int]]。
- **ValueError** - 如果 `kernel_size``dilation``stride` 值小于等于0或者个数大于2。
- **ValueError** - 如果 `padding_value` 值小于0或者个数大于2。
- **ValueError** - 如果 `input_x.dims(2)` 不等于 `kernel_size[0] * kernel_size[1]`
- **ValueError** - 如果 `input_x.dims(3)` 与计算出的滑动块数量不匹配。