批量点产品与纽比?

我需要用一个向量得到多个向量的点积。示例代码:

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

b = np.array([
    [0, 1, 2],
    [4, 5, 6],
    [-1, 0, 1],
    [-3, -2, 1]
])

我想得到每行的点积 b 反对 a .我可以迭代:

result = []
for row in b:
    result.append(np.dot(row, a))

print(result)

它给出:

[5, 17, 2, 0]

我怎么能不用迭代就得到这个呢?谢谢!