r - 如何将向量与自身进行 n 次卷积

Question

假设我有一个向量 x，我想对其进行 n 次卷积。在 R 中执行此操作的好方法是什么？

假设我们已经有一个函数 conv(u,v) 可以对两个向量进行卷积。

我能做到:

autoconv<-function(x,n){
    r<-1;
    for(i in 1:n){
        r<-conv(r,x);
    }
    return(r);
}

有没有更高效的方法？

Answer 1

对 x 进行快速傅里叶变换 (fft)，将其提高到 k 次方，然后对 fft 求逆。然后将其与对 x 的 k 副本执行卷积进行比较。没有使用包。

# set up test data
set.seed(123)
k <- 3 # no of vectors to convolve
n <- 32 # length of x
x <- rnorm(n)

# method 1 using fft and inverse fft
yy <- Re(fft(fft(x)^k, inverse = TRUE) / n)

# method 2 using repeated convolutions
y <- x
if (k >= 2) for(i in 2:k) y <- convolve(x, y, FALSE)

# check that the two methods give the same result
all.equal(y, yy)
## TRUE