gradient_utils.h

Go to the documentation of this file.

#ifndef SOPT_GRADIENT_UTILS_H
#define SOPT_GRADIENT_UTILS_H
 
#include "sopt/types.h"
#include "sopt/linear_transform.h"
#include <memory>
 
namespace sopt {
 
template <typename T>
class IterationState {
  public:
  IterationState(const T& target)
  {
    _Phi = std::make_shared<sopt::LinearTransform<T>>(linear_transform_identity<T>());
  }
 
  IterationState(const T& target,
                 std::shared_ptr<sopt::LinearTransform<T>> Phi)
                 : _target(target) {
    _Phi = Phi;
  }
 
  const T& target() const { return _target; }
 
  const sopt::LinearTransform<T>& Phi() const { return *_Phi; }
 
  void Phi(const sopt::LinearTransform<T> &new_phi)
  {
    _Phi = std::make_shared<sopt::LinearTransform<T>>(new_phi);
  }
 
  private:
  const T _target;
 
  std::shared_ptr<sopt::LinearTransform<T>> _Phi;
};
 
} // namespace sopt
#endif