ありがとう！Pytorchでのカスタムニューラルネットの実装を最適化しようとしています。2つの機能のどれが2つの操作のより効率的な実装になるかを知っていますか？一方の関数が他の関数よりも速いかどうか知りたいです。
これらの操作のどれがより効率的であるかを言うのは困難です。これは、関係するテンソルのサイズと形状、基礎となるハードウェア、Pytorchが作成した最適化などのさまざまな要因に依存するためです。

一般に、TORCH.BMM関数はマトリックスの乗算に最適化されているため、この特定の操作のアインズサム関数よりも速くなる可能性があります。ただし、北部は一般に柔軟性が高く、マトリックス増殖だけでなく、幅広いテンソル操作を実行するために使用できます。