torch.mul() 和 torch.mm() 区别
摘要：
torch.mul(a, b)是矩阵a和b对应位相乘，a和b的维度必须相等，比如a的维度是(1, 2)，b的维度是(1, 2)，返回的仍是(1, 2)的矩阵torch.mm(a, b)是矩阵a和b矩阵相乘，比如a的维度是(1, 2)，b的维度是(2, 3)，返回的就是(1, 3)的矩阵 阅读全文
posted @ 2021-03-15 17:23 cltt 阅读(552) 评论(0) 推荐(0) 编辑
