其他分享
首页 > 其他分享> > pytorch 记录

pytorch 记录

作者:互联网

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

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

矩阵相乘

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

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

标签:10,相乘,torch,矩阵,广播,bmm,pytorch,记录
来源: https://www.cnblogs.com/jisangday1/p/16136632.html