pytorch 中的矩阵乘法

参考 『PyTorch』矩阵乘法总结

1. *

两个张量在对应的位置上进行数值相乘。

x = torch.randn(2, 2)
y = x * x

2. torch.mm()

二维矩阵乘法

x = torch.randn(2, 2)
y = torch.mm(x, x)
# 或
y = x.mm(x)