matmul函数

torch.matmul(input, other, out = None)

函数对 input 和 other 两个张量进行矩阵相乘。torch.matmul 函数根据传入参数的张量维度有很多重载函数。

 

posted @ 2022-01-25 17:54  Tomorrow1126  阅读(945)  评论(0编辑  收藏  举报