Question

我对以下问题有疑问：

假设我有一些矩阵

A1 <- matrix(runif(rowsA1*T), rowsA1, T)
…
AD <- matrix(runif(rowsAD*T), rowsAD, T)

矩阵的数量是可变的（但肯定不是太大）。有没有办法更有效地执行以下操作（但在允许可变数量的矩阵的设置中）：

f1 <- function(A1, A2, ..., AD) {
  for(i in 1:nrow(A1)) {
    for(j in 1:nrow(A2)) {
      ...
        for(d in 1:nrow(AD)) {
          ret[i,j,...,d] <- \sum_{t=1}^T (A1[i,t]*A2[j,t]*...*AD[d,t])
        }
      ...
    }
  }
ret
}

非常感谢你的帮助！

罗曼

----------------------------------用例子编辑---------- ------------------------

 A1 <- |a b c|  A2 <- |j k l|  A3 <- |s t u|
       |d e f|        |m n o|        |v w x|
       |g h i|        |p q r|        |y z ä|

我想要获得以下内容：

ret[1,1,1] <- a*j*s + b*k*t + c*l*u
ret[2,1,3] <- d*j*y + e*k*z + f*l*ä

希望这使我的观点更加清晰。

----------------------------------编辑2013年11月26日------- ------------------------

你好@flodel。我试图实现你的代码，但是一旦有三个以上的矩阵就会出现问题。

假设我有以下矩阵

A1 <- matrix(runif(4*3), nrow = 4, ncol = 3)
A2 <- matrix(runif(3*3), nrow = 3, ncol = 3)
A3 <- matrix(runif(2*3), nrow = 2, ncol = 3)
A4 <- matrix(runif(1*3), nrow = 1, ncol = 3)

并将其插入您的代码

output.f1 <- f1(A1,A2,A3,A4)

提供正确的维度数

dim(output)
# [1] 4 3 2 1

但输出充满了NA

output.f1
# , , 1, 1

           # [,1] [,2] [,3]
# [1,] 0.13534704   NA   NA
# [2,] 0.07360135   NA   NA
# [3,] 0.07360135   NA   NA
# [4,] 0.07360135   NA   NA

# , , 2, 1

     # [,1] [,2] [,3]
# [1,]   NA   NA   NA
# [2,]   NA   NA   NA
# [3,]   NA   NA   NA
# [4,]   NA   NA   NA

感谢您的帮助...

最佳，罗曼

Answer 1

试一试。使用大apply循环时，对于大型矩阵，它可能会很慢，但只要对任意数量的矩阵都是通用的，它就不会有相同的行数：

f1 <- function(...) {
  args    <- list(...)
  nrows   <- sapply(args, nrow)
  idx     <- do.call(expand.grid, lapply(nrows, seq.int))
  get.row <- function(i, mat) mat[i, ]
  get.val <- function(i.vec) sum(Reduce(`*`, Map(get.row, i.vec, args)))
  idx$val <- apply(idx, 1, get.val)
  ret     <- array(NA, dim = nrows)
  ret[as.matrix(idx[, seq_along(args)])] <- idx$val
  ret
}

使用示例：

A1 <- matrix(1:12, nrow = 4, ncol = 3)
A2 <- matrix(1:9,  nrow = 3, ncol = 3)
A3 <- matrix(1:6,  nrow = 2, ncol = 3)

out <- f1(A1, A2, A3)

检查：

identical(out[3, 2, 1],
          sum(A1[3, ] * A2[2, ] * A3[1, ]))
# [1] TRUE

动态更高阶的逐行产品

1 个答案: