使用 apply 函数获取 R 中数组的乘积之和

Question

假设我有两个三维数组，我想要基于其中一个索引的数组乘积的总和。我想要的是下面示例代码最后一行中的总和。我知道我可以使用循环，但我想以一种有效的方式做到这一点，希望有一些 R 函数可以做这样的事情。任何帮助将不胜感激。

a <- array(1:12, dim=c(3, 2, 2))
b <- array(1, dim=c(3, 2, 2))
a[1, , ] %*% t(b[1, , ]) + a[2, , ] %*% t(b[2, , ]) + a[3, , ] %*% t(b[3, , ])

Answer 1

除非您确实遇到严重的低效率问题，否则请使用 for 循环。 你不能真正在两个对象上使用内置的 apply。（见评论）

注意apply isn't guaranteed to be faster than regular for loops。

编辑：由于 cmets：

Reduce(`+`, lapply(1:dim(a)[1], function(i) a[i, , ] %*% t(b[i, , ])))

是一个潜在的解决方案。虽然我怀疑它比直截了当更有效

sum <- matrix(0, ncol = dim(a)[2], nrow =  dim(a)[2])
for (i in 1: dim(a)[1]) sum <- sum + a[i, , ] %*% t(b[i, , ])

我认为它试图做的事情要清楚得多。