2016-10-01 13 views
-1

n×n対称性のテプリッツ行列T、長さnのベクトルvがあり、行列ベクトル積T%*%vを素早く計算したいと考えています。高速フーリエ変換を使用してT%*%v(または他の方法があれば)の変換方法を使用できるパッケージがRにありますか?たとえば、MatlabにはToeplitzmultパッケージがあります。テプリッツ行列のベクトル乗算R

答えて

1

以下の機能が機能します。 ifft()関数にはpracmaライブラリが必要です。サイズ1000のベクトル/行列について

toepmult <- function(A,v){ 
    n <- nrow(A) 
    x <- as.matrix(c(A[1,],0,A[1,][n:2])) 
    p <- c(v,rep(0,n)) 
    h <- as.vector(fft(p)*fft(x)) 
    out <- Re(pracma::ifft(h)[1:n]) 
    return(matrix(out,n)) 
} 

toepmult関数はA%*%vにかかる時間の約18%を取ります。サイズ10,000ベクトル/行列について

A <- toeplitz(runif(1000)) 
v <- runif(1000) 
microb(A%*%v,toepmult(A,v),times=1000) 
#Unit: microseconds 
#   expr  min  lq  mean median  uq  max neval 
#  A %*% v 1515.858 1597.345 1809.3517 1693.533 1957.4350 3868.788 1000 
# toepmult(A, v) 185.901 215.395 331.2928 298.435 347.7335 4611.285 1000 
#[[1]] 
#   [,1]  [,2] 
#median 1693.533 298.435 
#ratio  1.000  0.176 
#diff  0.000 -1395.098 

toepmult関数はA%*%vにかかる時間の約2.5%を要します。

A <- toeplitz(runif(10000)) 
v <- runif(10000) 
microb(A%*%v,toepmult(A,v),times=1000) 
#Unit: milliseconds 
#   expr  min   lq  mean  median   uq  max neval 
#  A %*% v 145.834304 160.395663 181.842779 170.396014 186.221449 495.2003 1000 
# toepmult(A, v) 2.802058 4.077408 4.990894 4.322707 4.911103 180.4926 1000 
#[[1]] 
#   [,1]  [,2] 
#median 170.396 4.323 
#ratio 1.000 0.025 
#diff  0.000 -166.073