Struct SGDParamState¶
Defined in File sgd.h
Page Contents
Inheritance Relationships¶
Base Type¶
public torch::optim::OptimizerCloneableParamState< SGDParamState >(Template Class OptimizerCloneableParamState)
Struct Documentation¶
-
struct SGDParamState : public torch::optim::OptimizerCloneableParamState<SGDParamState>¶
Public Functions
-
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¶
-
virtual void serialize(torch::serialize::InputArchive &archive) override¶
-
virtual void serialize(torch::serialize::OutputArchive &archive) const override¶
Friends
-
friend bool operator==(const SGDParamState &lhs, const SGDParamState &rhs)¶
-
inline auto momentum_buffer(const torch::Tensor &new_momentum_buffer) -> decltype(*this)¶
