如何计算R中矩阵乘积的对角线

我有两个矩阵A and B，那么最快的方法是什么just计算diag(A%*%B)，即内积ith row of A and ith的列B, and 不关心其他项的内积.

补充：A and B分别具有较大的行数和列数。

这可以在不进行完整矩阵乘法的情况下完成，仅使用矩阵元素的乘法。

我们需要将行相乘A通过匹配的列B并对元素求和。行数A是列t(A)，我们按元素乘以B并对列求和。

换句话说：colSums(t(A) * B)

测试代码我们首先创建示例数据：

n = 5
m = 10000;

A = matrix(runif(n*m), n, m);
B = matrix(runif(n*m), m, n);

你的代码：

diag(A %*% B)
# [1] 2492.198 2474.869 2459.881 2509.018 2477.591

无需矩阵乘法直接计算：

colSums(t(A) * B)
# [1] 2492.198 2474.869 2459.881 2509.018 2477.591

结果是一样的。