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 或 None) – 用于存放输出的数组。

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

  • cval (标量) – 如果模式为 constant，则用于填充输入边缘以外区域的值。默认为 0.0。

  • origin (标量 或 标量元组) – origin 参数控制滤波器相对于当前输入元素中心的放置位置。默认值 0 等同于 (0,)*input.ndim。

返回:

卷积结果。

返回类型:

cupy.ndarray

另请参阅

scipy.ndimage.convolve()

注意

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