autogradモジュール

ウィキペディア小見出し辞書

autogradモジュール

出典: フリー百科事典『ウィキペディア（Wikipedia）』 (2022/03/22 18:50 UTC 版)

「PyTorch」の記事における「autogradモジュール」の解説

PyTorchは自動微分と呼ばれるメソッドを利用する。recorderは実行された演算を記録し、その後、勾配の計算を行うときに記録した演算を逆方向にリプレイする。このメソッドは、ニューラルネットワークの構築時に特に強力であるため、順方向のパラメータの微分を計算することで、1エポックの計算にかかる時間を節約することができる。