nnモジュール
出典: フリー百科事典『ウィキペディア(Wikipedia)』 (2022/03/22 18:50 UTC 版)
PyTorchのautogradは、簡単に計算グラフを定義して勾配を得られるようになっているが、生のautogradは複雑なニューラルネットワークを定義するにはすこし低レベルすぎる場合がある。そのような場合のサポートとしてnnモジュールが提供されている。
※この「nnモジュール」の解説は、「PyTorch」の解説の一部です。
「nnモジュール」を含む「PyTorch」の記事については、「PyTorch」の概要を参照ください。
- nnモジュールのページへのリンク