pytorch:torch.mm()和torch.matmul()

微信公众号:数据挖掘和分析学习微信 torch.mm(mat1, mat2, out=None) → Tensor torch.matmul(mat1, mat2, out=None) → Tensor 对矩阵mat1和mat2进行相乘。 若是mat1 是一个n×m张量,mat2 是一个 m×p 张量,将会输出一个 n×p 张量out。学习 参数 :spa mat1 (Tensor) – 第一个相
相关文章
相关标签/搜索