cupyx.scipy.ndimage.convolve#

cupyx.scipy.ndimage.convolve(input, weights, output=None, mode='reflect', cval=0.0, origin=0)[源代码][源代码]#

多维卷积。

数组与给定的核进行卷积。

参数:

  • input (cupy.ndarray) – 输入数组。

  • weights (cupy.ndarray) – 权重数组，维度与输入相同

  • output (cupy.ndarray, dtype or None) – 要放置输出的数组。

  • mode (str) – 数组边界根据给定的模式 ('reflect', 'constant', 'nearest', 'mirror', 'wrap') 进行处理。默认是 'reflect'。

  • cval (scalar) – 如果模式是 constant，则在输入边缘之外填充的值。默认值是 0.0。

  • origin (scalar or tuple of scalar) – origin 参数控制过滤器相对于输入当前元素中心的放置位置。默认值 0 等同于 (0,)*input.ndim。

返回:

卷积的结果。

返回类型:

cupy.ndarray

参见

scipy.ndimage.convolve()

备注

当输出数据类型为整数（或当未提供输出且输入为整数时），由于中间结果的浮点数舍入，结果可能不完全匹配 SciPy 的结果。