SquaredExponentialCovariance.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 "CovarianceFunctionBase.h"

class SquaredExponentialCovariance : public CovarianceFunctionBase
{
public:
  static InputParameters validParams();
  SquaredExponentialCovariance(const InputParameters & parameters);

  void computeCovarianceMatrix(torch::Tensor & K,
                               const torch::Tensor & x,
                               const torch::Tensor & xp,
                               const bool is_self_covariance) const override;

  static void SquaredExponentialFunction(torch::Tensor & K,
                                         const torch::Tensor & x,
                                         const torch::Tensor & xp,
                                         const torch::Tensor & length_factor,
                                         const torch::Tensor & sigma_f_squared,
                                         const torch::Tensor & sigma_n_squared,
                                         const bool is_self_covariance);

  bool computedKdhyper(torch::Tensor & dKdhp,
                       const torch::Tensor & x,
                       const std::string & hyper_param_name,
                       unsigned int ind) const override;

  static void computedKdlf(torch::Tensor & K,
                           const torch::Tensor & x,
                           const torch::Tensor & length_factor,
                           const torch::Tensor & sigma_f_squared,
                           const int ind);

protected:
  const torch::Tensor & _length_factor;

  const torch::Tensor & _sigma_f_squared;

  const torch::Tensor & _sigma_n_squared;
};

#endif