numpy矩阵向量乘法

Question

当我将大小为（nxn）*（nx 1）的两个numpy数组相乘时，会得到大小为（nxn）的矩阵。遵循正常的矩阵乘法规则，期望（nx 1）向量，但我根本无法在Python的Numpy模块中找到有关此操作的任何信息。

事实是，我不想手动实现它以保持程序的速度。

示例代码如下所示：

a = np.array([[ 5, 1 ,3], [ 1, 1 ,1], [ 1, 2 ,1]])

b = np.array([1, 2, 3])

print a*b

   >>

   [[5 2 9]

   [1 2 3]

   [1 4 3]]

我想要的是：

print a*b

   >>

   [16 6 8]