cupyx.scipy.signal.convolve#

cupyx.scipy.signal.convolve(in1, in2, mode='full', method='auto')[源代码][源代码]#

卷积两个 N 维数组。

将 in1 和 in2 进行卷积，输出大小由 mode 参数决定。

参数:

  • in1 (cupy.ndarray) – 第一个输入。

  • in2 (cupy.ndarray) – 第二个输入。应与 in1 具有相同的维度。

  • mode (str) – 指示输出的大小： - 'full'：输出是完整的离散线性卷积（默认） - 'valid'：输出仅包含那些不依赖于零填充的元素。in1 或 in2 在每个维度上都必须至少与另一个一样大。 - 'same'： - 输出与 in1 大小相同，相对于 'full' 输出居中

  • method (str) – 指示用于计算的方法： - 'direct'：卷积直接通过求和确定，即卷积的定义 - 'fft'：通过调用 fftconvolve 使用傅里叶变换来执行卷积 - 'auto'：根据对参数的估计自动选择直接或FFT，以确定哪个更快（默认）。

返回:

卷积的结果。

返回类型:

cupy.ndarray

参见

cupyx.scipy.signal.choose_conv_method()

参见

cupyx.scipy.signal.correlation()

参见

cupyx.scipy.signal.fftconvolve()

参见

cupyx.scipy.signal.oaconvolve()

参见

cupyx.scipy.ndimage.convolve()

参见

scipy.signal.convolve()

备注

默认情况下，convolve 和 correlate 使用 method='auto'，它会调用 choose_conv_method 来选择使用预计算值的最快方法。CuPy 可能不会选择与 SciPy 在相同输入下计算卷积相同的方法。