逐行将两个2d数组相乘的函数

Question

我需要一个简单快速的函数来将numpy数组'a'的每一行乘以数组'b' a，b具有相同的二维尺寸 像这样的结果是example（c）： 但我想要一个插入此循环的numpy函数

a=np.arange(6).reshape(3,2)
b=np.arange(6,12).reshape(3,2)
c=np.array([[a[i,:]@b[i,:]]for i in range(a.shape[0])])

Answer 1

一种简单的方法是使用向量化numpy方法自己编写：

np.sum(a*b,axis=1,keepdims=True)

array([[  8],
       [ 48],
       [104]])