python - 三维数组作为向量和矩阵的乘法

Question

我有一个数组A(shape = (a, 1))和矩阵B(shape = (b1, b2))。想要将后者与前者的每个元素相乘，生成一个三维数组(shape = (a, b1, b2))。

有矢量化的方法来做到这一点吗？

import numpy as np
A = np.random.rand(3, 1)
B = np.random.rand(5, 4)
C = np.array([ a * B for a in A ])

Answer 1

有多种方法可以实现这一目标。一种是使用np.dot ，请注意，有必要在 B 中引入第二个轴，以便两个 ndarrays 可以相乘:

C = np.dot(A,B[:,None])
print(C.shape)
# (3, 5, 4)

使用np.multiply.outer ，正如 @divakar 所建议的:

C = np.multiply.outer(A,B)
print(C.shape)
# (3, 5, 4)

或者您也可以使用np.einsum :

C = np.einsum('ij,kl->ikl', A, B)
print(C.shape)
# (3, 5, 4)