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

25 lines
1.5 KiB
ReStructuredText
Raw Normal View History

mindspore.ops.col2im
====================
.. py:function:: mindspore.ops.col2im(input_x, output_size, kernel_size, dilation, padding_value, stride)
将一组滑动局部块组合成一个大的张量。
2022-07-25 15:09:45 +08:00
参数:
- **input_x** (Tensor) - 4维Tensor输入的批量的滑动局部块数据类型支持float16和float32。
- **output_size** (Tensor) - 包含两个int元素的1维Tensor输出张量的后两维的shape。
- **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)` 与计算出的滑动块数量不匹配。