使用星型运算符的矩阵乘法

Question

我很好奇调用 torch.mm(A, B) 和 A*B 之间有什么区别？

看起来 torch.mm 给了我们理想的结果，但 A*B 有时不起作用。

如果能提供一些文档就更好了。

谢谢你！

Answer 1

torch.mm(A,B)是常规矩阵乘法并且A*B是逐元素乘法。你可以在这个讨论中阅读它。对于矩阵乘法，@如果我没记错的话，你也可以使用。