一維傅里葉卷積 python

Question

問題

我想使用傅里葉變換編寫一個非常簡單的一維卷積。 我的代碼沒有給出預期的結果。

代碼

import numpy as np
import scipy
def fftconvolve(x, y):
    ''' Perso method to do FFT convolution'''
    fftx = np.fft.fft(x)
    ffty = np.fft.fft(y)
    fftc = fftx * ffty
    c = np.fft.ifft(fftc)
    return c.real

square = [0,0,0,1,1,1,0,0,0,0] # Example array

output = fftconvolve(square, square)
output2 = scipy.signal.fftconvolve(square, square, mode='same')

結果

圖展示了使用 scipy（橙色）和我的 function（藍色）的卷積。

問題

為什么兩個輸出不同？

Answer 1

如果您使用“完整”模式而不是“相同”模式，您會更接近！

import numpy as np
import scipy
def fftconvolve(x, y):
    ''' Perso method to do FFT convolution'''
    fftx = np.fft.fft(x)
    ffty = np.fft.fft(y)
    fftc = fftx * ffty
    c = np.fft.ifft(fftc)
    return c.real

square = [0,0,0,1,1,1,0,0,0,0] # Example array

output = fftconvolve(square, square)
output2 = scipy.signal.fftconvolve(square, square, mode='full')