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