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 "Control.h"
 
class LibtorchNeuralNetControl : public Control
{
public:
  static InputParameters validParams();
 
  LibtorchNeuralNetControl(const InputParameters & parameters);
 
  virtual void execute() override;
 
  Real getSignal(const unsigned int signal_index) const;
 
  unsigned int numberOfControlSignals() const { return _control_names.size(); }
 
  void loadControlNeuralNet(const Moose::LibtorchArtificialNeuralNet & input_nn);
 
  const Moose::LibtorchNeuralNetBase & controlNeuralNet() const;
 
  bool hasControlNeuralNet() const { return (_nn != NULL); };
 
protected:
  void conditionalParameterError(const std::string & param_name,
                                 const std::vector<std::string> & conditional_param,
                                 bool should_be_defined = true);
 
  void updateCurrentResponse();
 
  torch::Tensor prepareInputTensor();
 
  std::vector<Real> _current_response;
  std::vector<std::vector<Real>> & _old_responses;
 
  const std::vector<std::string> & _control_names;
  std::vector<Real> _current_control_signals;
 
  const std::vector<PostprocessorName> & _response_names;
 
  std::vector<const Real *> _response_values;
 
  const unsigned int _input_timesteps;
 
  const std::vector<Real> _response_shift_factors;
  const std::vector<Real> _response_scaling_factors;
  const std::vector<Real> _action_scaling_factors;
 
  std::shared_ptr<Moose::LibtorchNeuralNetBase> _nn;
};
 
#endif