Rate this Page
★★★★★

Struct AdamParamState#

Inheritance Relationships#

Base Type#

Struct Documentation#

structAdamParamState:publictorch::optim::OptimizerCloneableParamState<AdamParamState>#

Public Functions

inlineautostep(constint64_t&new_step)->decltype(*this)#
inlineautostep(int64_t&&new_step)->decltype(*this)#
inlineconstint64_t&step()constnoexcept#
inlineint64_t&step()noexcept#
inlineautoexp_avg(consttorch::Tensor&new_exp_avg)->decltype(*this)#
inlineautoexp_avg(torch::Tensor&&new_exp_avg)->decltype(*this)#
inlineconsttorch::Tensor&exp_avg()constnoexcept#
inlinetorch::Tensor&exp_avg()noexcept#
inlineautoexp_avg_sq(consttorch::Tensor&new_exp_avg_sq)->decltype(*this)#
inlineautoexp_avg_sq(torch::Tensor&&new_exp_avg_sq)->decltype(*this)#
inlineconsttorch::Tensor&exp_avg_sq()constnoexcept#
inlinetorch::Tensor&exp_avg_sq()noexcept#
inlineautomax_exp_avg_sq(consttorch::Tensor&new_max_exp_avg_sq)->decltype(*this)#
inlineautomax_exp_avg_sq(torch::Tensor&&new_max_exp_avg_sq)->decltype(*this)#
inlineconsttorch::Tensor&max_exp_avg_sq()constnoexcept#
inlinetorch::Tensor&max_exp_avg_sq()noexcept#
virtualvoidserialize(torch::serialize::InputArchive&archive)override#
virtualvoidserialize(torch::serialize::OutputArchive&archive)constoverride#

Friends

friendbooloperator==(constAdamParamState&lhs,constAdamParamState&rhs)#