行和列的点积

2024-04-26 14:38:29 发布

您现在位置:Python中文网/ 问答频道 /正文

我有两个矩阵,a(mxn)和b(nxp)。对于每个n,我想用a的第n列乘以b的第n行,得到n(mxp)矩阵。然后,我想通过计算矩阵中每个元素的平均值,将这些矩阵“折叠”成单个(mxp)矩阵。在numpy中有没有一种合理有效的方法(比如在no循环中)?在


Tags: 方法nonumpy元素矩阵平均值nxpmxn
1条回答
网友
1楼 · 发布于 2024-04-26 14:38:29

点积做乘法后加和,产生一个(m,p)数组。如果你想要的是平均值而不是求和,只需除以n,你要求和的项数。在

np.dot(a,b)/n

相关问题 更多 >