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]])