假设我们有两个矩阵A,B并且让矩阵C为A*B(矩阵乘法不是元素).我们希望只获得对角线条目C,可以通过np.diagonal(C).然而,这会导致不必要的时间开销,因为我们将A与B相乘,即使我们只需要每行的乘法A与B具有相同"id"的列,即行的第1 行,第A1 B行2的第2 A列B等等:形成对角线的乘法C.有没有办法使用Numpy有效地实现这一目标？我想避免使用循环来控制哪一行与哪一列相乘,相反,我希望有一个内置的numpy方法来执行这种操作来优化性能.
提前致谢..