einops 0.4 : PyTorch サンプル : pytorch と einops でより良いコードを書く (翻訳/解説)

【einops 0.4 : PyTorch と einops でより良いコードを書く】
einops は可読性と信頼性の高いコードのための柔軟でパワフルな tensor 演算です。numpy, pytorch, tensorflow, jax 等をサポートしています。0.4 をベースに幾つかドキュメントを翻訳しています。

今回は PyTorch サンプル「pytorch と einops でより良いコードを書く」です。現実的な例で、コードを改良する方法と einops がどのように役立てるかを学習します。

単純な畳込みモデルから最新の Transformer モデルまで、深層学習のビルディングブロックを書き直します。これらのコード断片は公式チュートリアルとポピュラーなレポジトリから抜粋されています。 [More]