Struct AdagradParamState

• Defined in File adagrad.h

Inheritance Relationships

Base Type

• public torch::optim::OptimizerCloneableParamState< AdagradParamState > (Template Class OptimizerCloneableParamState)

Struct Documentation

struct AdagradParamState : public torch::optim::OptimizerCloneableParamState<AdagradParamState>

Public Functions

inline auto sum(const torch::Tensor &new_sum) -> decltype(*this)

inline auto sum(torch::Tensor &&new_sum) -> decltype(*this)

inline const torch::Tensor &sum() const noexcept

inline torch::Tensor &sum() noexcept

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

virtual void serialize(torch::serialize::InputArchive &archive) override

virtual void serialize(torch::serialize::OutputArchive &archive) const override

Friends

friend bool operator==(const AdagradParamState &lhs, const AdagradParamState &rhs)