LMC.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 LMC : public CovarianceFunctionBase
{
public:
  static InputParameters validParams();
  LMC(const InputParameters & parameters);

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

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

protected:
  void computeBMatrix(torch::Tensor & Bmat, const unsigned int exp_i) const;

  void
  computeAGradient(torch::Tensor & grad, const unsigned int exp_i, const unsigned int index) const;

  void computeLambdaGradient(torch::Tensor & grad,
                             const unsigned int exp_i,
                             const unsigned int index) const;

  const unsigned int _num_expansion_terms;

private:
  std::vector<const torch::Tensor *> _a_coeffs;
  std::vector<const torch::Tensor *> _lambdas;
};

#endif