- 博客(2)
- 收藏
- 关注
RSS订阅原创 torch.mm()和torch.matmul()
torch.mm()和都是PyTorch库中用来进行矩阵乘法的函数，但它们在处理输入时有一些不同。torch.mm()：这个函数仅接受二维矩阵作为输入，并进行矩阵乘法。如果输入的张量不是二维的，它会抛出一个错误。：这个函数可以接受高于二维的张量，并进行适当的广播和矩阵乘法。对于二维矩阵，和torch.mm()的行为是相同的。# 张量a的第0维的大小是3，而张量b的第0维的大小是2，这两者不匹配，都不是1，不能广播。
2023-12-08 18:51:40
562
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人