Added directory torch_modules

CMakeLists.txt

@@ -25,10 +25,12 @@ set(CMAKE_CXX_FLAGS_RELEASE "-Ofast")
 include_directories(fmt/include)
 include_directories(common/include)
 include_directories(reading_machine/include)
+include_directories(torch_modules/include)
 include_directories(utf8)
 
 add_subdirectory(fmt)
 add_subdirectory(common)
 add_subdirectory(dev)
 add_subdirectory(reading_machine)
+add_subdirectory(torch_modules)
 

reading_machine/include/Classifier.hpp

@@ -2,6 +2,7 @@
 #define CLASSIFIER__H
 
 #include <string>
+#include <torch/torch.h>
 #include "TransitionSet.hpp"
 
 class Classifier

torch_modules/CMakeLists.txt

+FILE(GLOB SOURCES src/*.cpp)
+
+add_library(torch_modules STATIC ${SOURCES})
+target_link_libraries(torch_modules "${TORCH_LIBRARIES}")

torch_modules/include/MLP.hpp

+#ifndef MLP__H
+#define MLP__H
+
+class MLPImpl
+{
+
+};
+
+#endif

torch_modules/src/MLP.cpp

+#include "MLP.hpp"