使用星算子的矩阵乘法

Question

我很好奇调用torch.mm（A，B）和A * B有什么区别？

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

提供任何文档会更好。

谢谢！

Answer 1

torch.mm(A,B)是规则的矩阵乘法，而A*B是元素方式的乘法。您可以在此discussion上阅读它。对于矩阵乘法，如果我也没记错的话，可以使用@。