pytorch 记录

2022/4/12 23:44:58

本文主要是介绍pytorch 记录,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

torch.take() 把张量化为一个行向量,取其中的元素

广播机制中0维和1维的轴是可广播的

矩阵相乘

*和torch.mul()是对应位置相乘
@和torch.mm()是矩阵相乘 torch.bmm()相比mm增加了batch运算但不能广播
torch.matmul()相比bmm更是增加了广播,泛用性强

torch.full([3,3],10) 将数字10填入3X3的矩阵中



这篇关于pytorch 记录的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程