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