matmul函数

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

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