矢量化标量乘以矩阵的和,其中缩放器是列表的一个元素



我正在尝试对以下进行矢量化

a = np.array([1,2])
b = np.array([[5,5],[5,5]])
target = 0
for _ in a:
target = target + _ * b

上面产生了一个2x2矩阵,其中所有条目都是15。如何通过矢量化实现这一点?我一直在尝试将a转换为两个2x2矩阵,但我在方面没有任何进展

感谢

这看起来像一个简单的:

b * a.sum()

输出:

array([[15, 15],
[15, 15]])

最新更新