torch:zero-grad

(torch:zero-grad tensor-or-module)

蓄積された勾配をクリアして (nil に戻して) 引数自身を返します。テンソルならそれ自身の勾配、モジュールなら torch:parameters が到達するすべてのパラメータの勾配です。torch:backward は蓄積 (+=) するため、学習ループではステップ間でモデルに対してこれを呼びます。