python - 矢量化代码时出现问题

Question

我很难做到这一点。我有两个 m x n 矩阵(A 和 B)，我需要将 A 的每一列乘以 B 中的行，以生成 m x (n*n) 矩阵。我想我的解释不是很清楚，所以我会发布一个例子:

A = 
[1 2 
 3 4]
B = 
[5 6
 7 8]

我希望拥有:

 [[5 6] [10 12]
  [21 24] [28 32]]

我可以使用for循环来做到这一点，但我想尽可能避免for。还使用 numpy 来完成所有这些，所有数据都存储为 np.array。

Answer 1

也许:

>>> A = np.array([[1,2],[3,4]])
>>> B = np.array([[5,6],[7,8]])
>>> (A * B[None, :].T).T
array([[[ 5,  6],
        [21, 24]],

       [[10, 12],
        [28, 32]]])

我们使用 None 为 B 添加额外的维度，并进行一些转置以获得正确的对齐方式。