matmul函數

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

函數對 input 和 other 兩個張量進行矩陣相乘。torch.matmul 函數根據傳入參數的張量維度有很多重載函數。