Struct RMSpropParamState#
Defined in File rmsprop.h
Page Contents
Inheritance Relationships#
Base Type#
public torch::optim::OptimizerCloneableParamState< RMSpropParamState >(Template Class OptimizerCloneableParamState)
Struct Documentation#
-
struct RMSpropParamState : public torch::optim::OptimizerCloneableParamState<RMSpropParamState>#
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 square_avg(const torch::Tensor &new_square_avg) -> decltype(*this)#
-
inline auto square_avg(torch::Tensor &&new_square_avg) -> decltype(*this)#
-
inline const torch::Tensor &square_avg() const noexcept#
-
inline torch::Tensor &square_avg() noexcept#
-
inline auto momentum_buffer(const torch::Tensor &new_momentum_buffer) -> decltype(*this)#
-
inline auto momentum_buffer(torch::Tensor &&new_momentum_buffer) -> decltype(*this)#
-
inline const torch::Tensor &momentum_buffer() const noexcept#
-
inline torch::Tensor &momentum_buffer() noexcept#
-
inline auto grad_avg(const torch::Tensor &new_grad_avg) -> decltype(*this)#
-
inline auto grad_avg(torch::Tensor &&new_grad_avg) -> decltype(*this)#
-
inline const torch::Tensor &grad_avg() const noexcept#
-
inline torch::Tensor &grad_avg() noexcept#
-
virtual void serialize(torch::serialize::InputArchive &archive) override#
-
virtual void serialize(torch::serialize::OutputArchive &archive) const override#
Friends
-
friend bool operator==(const RMSpropParamState &lhs, const RMSpropParamState &rhs)#
-
inline auto step(const int64_t &new_step) -> decltype(*this)#