仅计算R中

Question

矩阵乘法的对角线我只需要在一个矩阵乘法对角线元素：

，

在R.

由于Z是巨大的我想避免全员出动乘... 。

Z <- matrix(c(1,1,1,2,3,4), ncol = 2) 
Z 
#  [,1] [,2] 
#[1,] 1 2 
#[2,] 1 3 
#[3,] 1 4 

X <- matrix(c(10,-5,-5,20), ncol = 2) 
X 
#  [,1] [,2] 
#[1,] 10 -5 
#[2,] -5 20 

Z %*% D %*% t(Z) 
#  [,1] [,2] [,3] 
#[1,] 70 105 140 
#[2,] 105 160 215 
#[3,] 140 215 290 

diag(Z %*% D %*% t(Z)) 
#[1] 70 160 290 

X始终是一个小方阵（2×2，3×3或4×4），其中Z将列等于X的维数是否有一个功能有效做到这一点？

Answer 1

我不认为你可以避开第一矩阵乘法（即ZX），但你可以在第二个，这是昂贵的：

rowSums((Z %*% X) * Z) 
# [1] 70 160 290 

第二乘法是不是一个矩阵乘法。这是更快：

library(microbenchmark) 
set.seed(1) 
X <- matrix(c(10,-5,-5,20), ncol = 2) 
Z <- matrix(sample(1:1000), ncol=2) # made Z a little bigger  

microbenchmark(
    res.new <- rowSums((Z %*% X) * Z), # this solution 
    res.old <- diag(Z %*% X %*% t(Z)) # original method 
) 
# Unit: microseconds 
#        expr  min  lq  mean median  uq  max neval 
# res.new <- rowSums((Z %*% X) * Z) 20.956 23.233 34.77693 29.6150 44.0025 67.852 100 
# res.old <- diag(Z %*% X %*% t(Z)) 571.214 699.247 1761.08885 760.4295 1188.4485 47488.543 100  

all.equal(res.new, res.old) 
# [1] TRUE