Shortcuts

Struct AdamParamState

Inheritance Relationships

Base Type

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)

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources