参考 『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)

如果觉得《[PyTorch] 矩阵乘法》对你有帮助，请点赞、收藏，并留下你的观点哦！