opm-common
List of all members | Public Member Functions
Opm::ML::NNLayerActivation< Evaluation > Class Template Reference
Inheritance diagram for Opm::ML::NNLayerActivation< Evaluation >:
Opm::ML::NNLayer< Evaluation >

Public Member Functions

 NNLayerActivation (ActivationType activation_type=ActivationType::kLinear)
 
bool loadLayer (std::ifstream &file) override
 
bool apply (const Tensor< Evaluation > &in, Tensor< Evaluation > &out) override
 
The documentation for this class was generated from the following file: