Shortcuts

Struct RMSpropParamState

Inheritance Relationships

Base Type

Struct Documentation

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

Public Functions

auto step(const int64_t &new_step) -> decltype(*this)
auto step(int64_t &&new_step) -> decltype(*this)
const int64_t &step() const noexcept
int64_t &step() noexcept
auto square_avg(const torch::Tensor &new_square_avg) -> decltype(*this)
auto square_avg(torch::Tensor &&new_square_avg) -> decltype(*this)
const torch::Tensor &square_avg() const noexcept
torch::Tensor &square_avg() noexcept
auto momentum_buffer(const torch::Tensor &new_momentum_buffer) -> decltype(*this)
auto momentum_buffer(torch::Tensor &&new_momentum_buffer) -> decltype(*this)
const torch::Tensor &momentum_buffer() const noexcept
torch::Tensor &momentum_buffer() noexcept
auto grad_avg(const torch::Tensor &new_grad_avg) -> decltype(*this)
auto grad_avg(torch::Tensor &&new_grad_avg) -> decltype(*this)
const torch::Tensor &grad_avg() const noexcept
torch::Tensor &grad_avg() noexcept
void serialize(torch::serialize::InputArchive &archive) override
void serialize(torch::serialize::OutputArchive &archive) const override
~RMSpropParamState() override = default

Friends

friend bool operator==(const RMSpropParamState &lhs, const RMSpropParamState &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