a = np.array([[1, 2], [3, 4], [5, 6]]) b = np.array([[2, 2], [3, 3], [4, 4]])
我想得到一个1d数组,在每个对应的索引处都有列表的点积,即。
[(1*2 + 2*2), (3*3 + 4*3), (5*4 + 6*4)] [6, 21, 44]
我该怎么做呢?提前谢谢!
你能做到的
np.sum(a*b,axis=1)