torch.matmul()用法介紹

torch.matmul是tensor的乘法，輸入可以是高維的。 當輸入是都是二維時，就是普通的矩陣乘法，和tensor.mm函數用法相同。 當輸入有多維時，把多出的一維作爲batch提出來，其他部分做矩陣乘法。 下面看一個兩個都是3維的例子。 將b的第0維1broadcast成2提出來，後兩維做矩陣乘法即可。 再看一個複雜一點的，是官網的例子。 首先把a的第0維2作爲batch提出來，則a和b都

>>阅读原文<<