Opm::ML::NNLayerActivation< Evaluation > Class Template Reference

Inheritance diagram for Opm::ML::NNLayerActivation< Evaluation >:

Public Types
enum class	ActivationType {   kLinear = 1 , kRelu = 2 , kSoftPlus = 3 , kSigmoid = 4 ,   kTanh = 5 , kHardSigmoid = 6 }

Public Member Functions
bool	loadLayer (std::ifstream &file) override
bool	apply (const Tensor< Evaluation > &in, Tensor< Evaluation > &out) override

Member Function Documentation

apply()

template<class Evaluation>

bool Opm::ML::NNLayerActivation< Evaluation >::apply ( const Tensor< Evaluation > & in, Tensor< Evaluation > & out )

overridevirtual

Implements Opm::ML::NNLayer< Evaluation >.

loadLayer()

template<class Evaluation>

bool Opm::ML::NNLayerActivation< Evaluation >::loadLayer ( std::ifstream & file )

overridevirtual

Implements Opm::ML::NNLayer< Evaluation >.

---

The documentation for this class was generated from the following files:

• opm/ml/ml_model.hpp
• opm/ml/ml_model.cpp