王冰冰

摘要： torch.mm 线代的矩阵乘法，要求输入都是矩阵 torch.matmul 注意：torch.mm和torch.matmul不等价 根据输入不同执行不同的操作： 输入都是二维矩阵，矩阵乘法，等同于torch.mm 输入都是一维向量，计算向量内积，等同于torch.dot 第一个参数是向量，第二个是 阅读全文