python - 2 个矩阵中行的点积

Question

所以我基本上有矩阵

a = [x1,y1,z1
     x2,y2 z2
     .....
     xN,yN,zN]

和b = [i1,j1,k1
i2,j2,k2
....]

并想取 a 的第 1 行与 b 的第 1 行的点积，a 的第 2 行与 b 的第 2 行的点积，依此类推。它们都是形状 (3,71216) 但是

np.dot(a,b) 

给出 ValueError:对象未对齐。任何帮助将不胜感激。

Answer 1

这是因为 np.dot 期望第一个参数的行数等于第二个参数的列数，如错误消息所示。

你需要对齐数组:

np.dot(a, b.T)

---

您的问题令人困惑，因为您说数组的形状为 (3, 71216)，但您的示例显示的是转置 (71216, 3)。

无论如何，听起来您只是想要这个，而不是点积:

(a * b).sum(axis=0)