torch.mm
torch.mm(input, mat2, *, out=None) → Tensor
执行矩阵input和mat2的矩阵乘法运算。
如果input是（n×m）张量，mat2是（m×p）张量，out将是（n x p）张量。
input（张量）–第一个要矩阵相乘的矩阵
mat2（张量）–第二个要矩阵相乘的矩阵
out（张量，可选）–输出张量。
>>> mat1 = torch.randn(2, 3)
>>> mat2 = torch.randn(3, 3)
>>> torch.mm(mat1, mat2)
tensor([[ 0.4851, 0.5037, -0.3633],[-0.0760, -3.6705, 2.4784]])