torch.mm只能让两个二维tensor作矩阵乘法torch.mul作element-wise的矩阵点乘,维数不限,可以矩阵乘标量torch.bmm作batch单位的矩阵乘法,维度只能为3 ...
確定! 回上一頁