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 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