一、torch.mm的基礎知識 torch.mm(input, mat2, out=None)函數是計算兩個tensor的矩陣乘法。其中，input是第一個矩陣，mat2是第二個矩…