Numpy 向量/矩阵乘法意外工作

Question

我的代码有问题：我想以如下所示的方式将两个“向量”相乘

# what i want
import numpy as np

a = np.array([1,2,3])
b = np.array([2,3,4,5])
z = f(a,b)

z -> array([
    [2,3,4,5],
    [4,6,8,10],
    [6,9,12,15]
])

f 应该如何定义？

Answer 1

您可以使用np.multiply.outer(a, b) 。

Answer 2

您可以在重塑的 arrays 上使用np.matmul function

a = np.reshape(a, (3,1))
b = np.reshape(b, (1,4))
np.matmul(a,b)