Logo ROOT  
Reference Guide
 
Loading...
Searching...
No Matches
ROperator_Selu.hxx
Go to the documentation of this file.
1#ifndef TMVA_SOFIE_ROPERATOR_SELU
2#define TMVA_SOFIE_ROPERATOR_SELU
3
5#include "TMVA/ROperator.hxx"
6#include "TMVA/RModel.hxx"
7
8#include <sstream>
9
10namespace TMVA{
11namespace Experimental{
12namespace SOFIE{
13
14template <typename T>
15class ROperator_Selu final : public ROperator
16{
17
18private:
19
20 std::string fNX;
21 std::string fNY;
22 std::vector<size_t> fShape;
23
24public:
26 ROperator_Selu(std::string nameX, std::string nameY):
27 fNX(UTILITY::Clean_name(nameX)), fNY(UTILITY::Clean_name(nameY)){}
28
29 std::vector<ETensorType> TypeInference(std::vector<ETensorType> input){
30 return input;
31 }
32
33 std::vector<std::vector<size_t>> ShapeInference(std::vector<std::vector<size_t>> input){
34 auto ret = input; //suggest copy to compiler
35 return ret;
36 }
37
38 void Initialize(RModel& model){
39 if (model.CheckIfTensorAlreadyExist(fNX) == false){ //input must be a graph input, or already initialized intermediate tensor
40 throw std::runtime_error("TMVA SOFIE Selu Op Input Tensor is not found in model");
41 }
42 fShape = model.GetTensorShape(fNX);
44 }
45
46
47 std::string Generate(std::string OpName){
48 OpName = "op_" + OpName;
49 if (fShape.empty()){
50 throw std::runtime_error("TMVA SOFIE Transpose Selu called to Generate without being initialized first");
51 }
52 std::stringstream out;
53 int length = 1;
54 for(auto& i: fShape){
55 length *= i;
56 }
57 out << "\t" << "for (int id = 0; id < " << length << " ; id++){\n";
58 out << "\t\t" << "tensor_" << fNY << "[id] = 1.0507009873554804934193349852946 * (std::max(float(0.0), tensor_" << fNX << "[id]) + std::min(0.0, 1.6732632423543772848170429916717 * (std::exp(" << "tensor_" << fNX << "[id]" <<")-1)));\n";
59 out << "\t}\n";
60 return out.str();
61 }
62
63};
64
65}//SOFIE
66}//Experimental
67}//TMVA
68
69
70#endif //TMVA_SOFIE_ROPERATOR_SELU
const ETensorType & GetTensorType(std::string name)
Definition RModel.cxx:70
void AddIntermediateTensor(std::string tensor_name, ETensorType type, std::vector< std::size_t > shape)
Definition RModel.cxx:136
bool CheckIfTensorAlreadyExist(std::string tensor_name)
Definition RModel.cxx:91
const std::vector< size_t > & GetTensorShape(std::string name)
Definition RModel.cxx:49
std::vector< std::vector< size_t > > ShapeInference(std::vector< std::vector< size_t > > input)
std::string Generate(std::string OpName)
ROperator_Selu(std::string nameX, std::string nameY)
std::vector< ETensorType > TypeInference(std::vector< ETensorType > input)
create variable transformations