矩阵按行相乘 [英] Matrix multiplication by row

问题描述

有什么方法可以更快地计算矩阵c吗?

Is there any way to calculate matrix c faster?

a=matrix(runif(10),2,5)
b=matrix(runif(15),3,5)

c=matrix(,nrow(a)*nrow(b),5)
k=0
for(i in 1:nrow(a)){
  for(j in 1:nrow(b)){
    k=k+1
    c[k,]=a[i,]*b[j,]
  }
}

推荐答案

这是我的版本:

c1 =  a[ rep(1:nrow(a), each = nrow(b)), ] * 
      b[ rep(1:nrow(b), times = nrow(a)), ];
all.equal(c, c1);

> TRUE

这篇关于矩阵按行相乘的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！