My Project
|
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 |
|
overridevirtual |
Implements Opm::ML::NNLayer< Evaluation >.
|
overridevirtual |
Implements Opm::ML::NNLayer< Evaluation >.