zl程序教程

您现在的位置是:首页 >  .Net

当前栏目

pytorch矩阵乘法总结

2023-02-18 16:32:06 时间

点乘 torch.mul(a,b)
二维矩阵乘 torch.mm(a,b)
三维矩阵乘 torch.bmm(a,b)
高维矩阵乘 torch.matmul(a,b)

https://blog.csdn.net/sunjinshengli/article/details/121901225