一、torch.mm的基础知识 torch.mm(input, mat2, out=None)函数是计算两个tensor的矩阵乘法。其中，input是第一个矩阵，mat2是第二个矩…