如何将 2d 矩阵添加到 3d 数组并保留 3d 数组的形状？

Question

我正在尝试向 3d 数组的每个“子矩阵”添加一个矩阵。我想在没有循环的情况下执行此操作。

a = array(1:24, dim = 2:4)
a[1,,]
a[2,,]

b = array(1:12, dim = 3:4)
b

c1 = a[1,,] + b
c2 = a[2,,] + b

c = apply(a, 1, function(a_){
  da_ = dim(a_)
  db = dim(b)
  message(sprintf("The dimensions of a_ are [%i x %i] and the dimensions of b are [%i x %i]", da_[1], da_[2], db[1], db[2]))
  a_ + b
})

在上面的代码中，我希望c[1,,] 等于c1，并且c[2,,] 等于c2。 这可以使用apply 函数吗？

PS：我找到了this similar question，但没有直接回答这个问题。

Answer 1

这是使用rep的一种方式：

d <- a + rep(b, each = dim(a)[1])
all.equal(d[1,,], c1)
# TRUE