pytorch笔记

 @,torch.matmul,torch.mm:矩阵相乘,第一个矩阵的列和第二个矩阵的行维度相同            *,torch.mul:矩阵对应元素相乘,所以两个矩阵维数相同,同维矩阵

torch.dot:一维的张量进行相乘再相加,结果是一个值

 

posted @ 2023-05-31 10:53  15375357604  阅读(6)  评论(0编辑  收藏  举报