PyTorchにおいて、テンソルの勾配計算が必要であることを指定するフラグはどれか。

requires_grad=Trueを設定することで、そのテンソルに対する演算が記録される。

関連記事

関連記事はありません。