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)


免责声明!

本站转载的文章为个人学习借鉴使用,本站对版权不负任何法律责任。如果侵犯了您的隐私权益,请联系本站邮箱yoyou2525@163.com删除。



 
粤ICP备18138465号  © 2018-2025 CODEPRJ.COM