pytorchmatmulとmmおよびbmmの違い

Difference Between Pytorch Matmul

pytorchのmatmulとmmおよびbmmの違い

• matmul
• んん
• bmm
• 結論として

まず、公式サイトでこれら3つの機能の紹介をご覧ください。

公式ドキュメントからわかるように、

• mmは行列の乗算のみを実行できます。つまり、入力の2つのテンソル次元は次のようになります。（n×m）（n times m）（（n××m）。と（m×p）（m times p）（（m××p）。
• bmmは、2つの3次元テンソルの乗算であり、2つのテンソル次元は次のとおりです。（b×m×p）（b times m times p）（（b××m××p）。と（b×n×p）（b times n times p）（（b××n××p）。、最初の次元bはbatch_sizeを表します
• Matmulはテンソル乗算を実行でき、入力は高次元にすることができます。