torch.matmul

 

0階張量是一個標量。

1階張量是一個向量。

2階張量是一個矩陣。

如果倆個張量都是一維度，輸出的結果是標量相乘。

如果倆個張量都是矩陣，輸出的結果是矩陣乘積。

如果第一個是二維，第二個是一維張量，輸出結果將第一個參數每一行看成一個向量與第二個進行向量積。（只有他需要注意）

如果第一個是一維張量，第二個是二維張量，輸出結果也是矩陣乘積。