Class OptimizerParamGroup

• Defined in File optimizer.h

Class Documentation

class OptimizerParamGroup

Stores parameters in the param_group and stores a pointer to the OptimizerOptions.

Public Functions

inline OptimizerParamGroup(const OptimizerParamGroup &param_group)

OptimizerParamGroup(OptimizerParamGroup &&param_group) = default

inline OptimizerParamGroup(std::vector<Tensor> params)

inline OptimizerParamGroup(std::vector<Tensor> params, std::unique_ptr<OptimizerOptions> options)

OptimizerParamGroup &operator=(const OptimizerParamGroup &param_group) = delete

OptimizerParamGroup &operator=(OptimizerParamGroup &&param_group) noexcept = default

~OptimizerParamGroup() = default

bool has_options() const

OptimizerOptions &options()

const OptimizerOptions &options() const

void set_options(std::unique_ptr<OptimizerOptions> options)

std::vector<Tensor> &params()

const std::vector<Tensor> &params() const

Protected Attributes

std::vector<Tensor> params_

std::unique_ptr<OptimizerOptions> options_