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