本文へ移動
10/7(水) 14:09
あとで読む
出典
視点Hugging Face Blog一次· 7/10(金)

PyTorch、attention の実装を比較

Hugging Face Blog
要

AI要約

PyTorch の attention 実装をプロファイラーで比較しました。 ナイーブ実装では masked_fill がメモリコピーを発生させます。 masked_fill_ に変えるとカーネルが 1 つ減ります。

AI要約です。詳しくは出典へ。誤りを報告

典

出典

1ソース · 一次情報
Hugging Face Blog一次Profiling in PyTorch (Part 3): Attention is all you profilehuggingface.co
出典を読む