LibtorchNeuralNetControl.h

Go to the documentation of this file.

//* This file is part of the MOOSE framework
//* https://mooseframework.inl.gov
//*
//* All rights reserved, see COPYRIGHT for full restrictions
//* https://github.com/idaholab/moose/blob/master/COPYRIGHT
//*
//* Licensed under LGPL 2.1, please see LICENSE for details
//* https://www.gnu.org/licenses/lgpl-2.1.html
 
#ifdef MOOSE_LIBTORCH_ENABLED
 
#pragma once
 
#include "LibtorchArtificialNeuralNet.h"
#include "LibtorchObservationHistoryHelper.h"
#include "Control.h"
 
class LibtorchNeuralNetControl : public Control
{
public:
  static InputParameters validParams();
 
  LibtorchNeuralNetControl(const InputParameters & parameters);
 
  virtual void initialSetup() override;
 
  virtual void execute() override;
 
  Real getSignal(const unsigned int signal_index) const;
 
  unsigned int numberOfControlSignals() const { return _control_names.size(); }
 
  virtual void loadControlNeuralNet(const Moose::LibtorchArtificialNeuralNet & input_nn);
 
  virtual void loadControlNeuralNetFromFile();
 
  const Moose::LibtorchNeuralNetBase & controlNeuralNet() const;
 
  bool hasControlNeuralNet() const { return _nn != nullptr; };
 
protected:
  void conditionalParameterError(const std::string & param_name,
                                 const std::vector<std::string> & conditional_param,
                                 bool should_be_defined = true);
 
  void updateCurrentObservation();
 
  torch::Tensor prepareInputTensor();
 
  std::vector<Real> _current_observation;
  std::vector<std::vector<Real>> & _old_observations;
 
  const std::vector<std::string> & _control_names;
  std::vector<Real> & _current_control_signals;
 
  const std::vector<PostprocessorName> & _observation_names;
 
  std::vector<const Real *> _observation_values;
 
  const unsigned int _input_timesteps;
 
  const std::vector<Real> _observation_shift_factors;
  const std::vector<Real> _observation_scaling_factors;
  const std::vector<Real> _action_scaling_factors;
 
  const LibtorchObservationHistoryHelper _observation_history;
 
  std::shared_ptr<Moose::LibtorchNeuralNetBase> _nn;
};
 
#endif