Struct AdamParamState¶
Defined in File adam.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::optim::OptimizerCloneableParamState< AdamParamState >
(Template Class OptimizerCloneableParamState)
Struct Documentation¶
-
struct AdamParamState : public torch::optim::OptimizerCloneableParamState<AdamParamState>¶
Public Functions
-
inline auto step(const int64_t &new_step) -> decltype(*this)¶
-
inline auto step(int64_t &&new_step) -> decltype(*this)¶
-
inline const int64_t &step() const noexcept¶
-
inline int64_t &step() noexcept¶
-
inline auto exp_avg(const torch::Tensor &new_exp_avg) -> decltype(*this)¶
-
inline auto exp_avg(torch::Tensor &&new_exp_avg) -> decltype(*this)¶
-
inline const torch::Tensor &exp_avg() const noexcept¶
-
inline torch::Tensor &exp_avg() noexcept¶
-
inline auto exp_avg_sq(const torch::Tensor &new_exp_avg_sq) -> decltype(*this)¶
-
inline auto exp_avg_sq(torch::Tensor &&new_exp_avg_sq) -> decltype(*this)¶
-
inline const torch::Tensor &exp_avg_sq() const noexcept¶
-
inline torch::Tensor &exp_avg_sq() noexcept¶
-
inline auto max_exp_avg_sq(const torch::Tensor &new_max_exp_avg_sq) -> decltype(*this)¶
-
inline auto max_exp_avg_sq(torch::Tensor &&new_max_exp_avg_sq) -> decltype(*this)¶
-
inline const torch::Tensor &max_exp_avg_sq() const noexcept¶
-
inline torch::Tensor &max_exp_avg_sq() noexcept¶
-
virtual void serialize(torch::serialize::InputArchive &archive) override¶
-
virtual void serialize(torch::serialize::OutputArchive &archive) const override¶
Friends
-
friend bool operator==(const AdamParamState &lhs, const AdamParamState &rhs)¶
-
inline auto step(const int64_t &new_step) -> decltype(*this)¶