python數組、矩陣相乘的多種方式

1. numpy.ndarray類型乘積

1.1 矩陣乘法

 

a@b

np.dot(a, b)

np.matmul(a, b)

1.2 對應位置元素相乘

a*b

np.multiply(a, b)

2.numpy.matrix類型乘積

2.1 矩陣乘法

a@b

a*b

np.dot(a,b)

np.matmul(a,b)

2.2 對應位置元素相乘

np.multiply(a,b)

3.torch.tensor類型乘積

3.1 矩陣乘法

torch.mm(a,b) == a.mm(b) # 不廣播，a、b必須都是二維張量
torch.matmul(a,b) == a.matmul(b) #廣播，a、b可能是一維張量、二維張量、多維張量，功能特別多

1.2 對應位置元素相乘

a*b

torch.mul(a,b) == a.mul(b)