糖尿病康复,内容丰富有趣,生活中的好帮手!
糖尿病康复 > pytorch笔记:pytorch的乘法

pytorch笔记:pytorch的乘法

时间:2021-03-11 22:39:04

相关推荐

pytorch笔记:pytorch的乘法

1 点乘

tensor与标量点乘的结果是每个元素乘以这个标量的值

tensor与行向量点乘的结果是每一列乘以对应的值

tensor与列向量点乘的结果是每一行乘以对应的值

矩阵进行点乘,就是对应位的乘积

点乘还有广播的特点(也是对应位的乘积)

2mul(与点乘相同)

tensor与标量mul的结果是每个元素乘以这个标量的值

tensor与行向量mul的结果是每一列乘以对应的值

tensor与列向量点乘的结果是每一行乘以对应的值

矩阵进行mul,就是对应位的乘积

mul也有广播的特点(也是对应位的乘积)

3 mm 矩阵乘法

4 matmul 矩阵乘法

a,b矩阵同3中的a,b矩阵

矩阵乘法的广播版本

5 bmm batch的乘法

计算两个tensor的矩阵乘法。

对于torch.bmm(a,b)中的两个参数,tensor a 的size为(b,h,w),tensor b的size为(b,w,h)。

注意两个tensor的维度必须为3.

比三维多,比三维少的tensor维度都不行。

二维的情况:

四维的情况

四维矩阵相乘,可以用matmul实现

bmm使用(作用理论上和matmul一样?甚至matmul功能还更强大一些?欢迎讨论!)

如果觉得《pytorch笔记:pytorch的乘法》对你有帮助,请点赞、收藏,并留下你的观点哦!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。