Standardizer.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 "MooseTypes.h"
#include <vector>
#include "DataIO.h"
 
#include "LibtorchUtils.h"
 
namespace StochasticTools
{
 
 
class Standardizer
{
public:
  Standardizer() = default;
 
  void set(const Real & n);
  void set(const Real & mean, const Real & stdev);
  void set(const Real & mean, const Real & stdev, const Real & n);
  void set(const std::vector<Real> & mean, const std::vector<Real> & stdev);
 
  const torch::Tensor & getMean() const { return _mean; }
  const torch::Tensor & getStdDev() const { return _stdev; }
 
  void computeSet(const torch::Tensor & input);
 
  void storeHelper(std::ostream & stream, void * context) const;
 
  void getStandardized(torch::Tensor & input) const;
 
  void getDestandardized(torch::Tensor & input) const;
 
  void getDescaled(torch::Tensor & input) const;
 
  void getScaled(torch::Tensor & input) const;
 
protected:
  torch::Tensor _mean;
  torch::Tensor _stdev;
};
 
} // StochasticTools namespace
 
template <>
void dataStore(std::ostream & stream, StochasticTools::Standardizer & standardizer, void * context);
template <>
void dataLoad(std::istream & stream, StochasticTools::Standardizer & standardizer, void * context);
 
#endif