Added SpatialGraph module, in its quite early version.

1 files changed, 5 insertions, 0 deletions

diff --git a/init.c b/init.c
index d4012e4..0f0936f 100644
--- a/init.c
+++ b/init.c
@@ -48,6 +48,9 @@ static const void* torch_DoubleTensor_id = NULL;
 #include "generic/Threshold.c"
 #include "THGenerateFloatTypes.h"
 
+#include "generic/SpatialGraph.c"
+#include "THGenerateFloatTypes.h"
+
 DLL_EXPORT int luaopen_libnnx(lua_State *L)
 {
   torch_FloatTensor_id = luaT_checktypename2id(L, "torch.FloatTensor");
@@ -66,6 +69,7 @@ DLL_EXPORT int luaopen_libnnx(lua_State *L)
   nn_FloatSpatialSparseCriterion_init(L);
   nn_FloatSpatialMSECriterion_init(L);
   nn_FloatSpatialClassNLLCriterion_init(L);
+  nn_FloatSpatialGraph_init(L);
 
   nn_DoubleSpatialLinear_init(L);
   nn_DoubleHardShrink_init(L);
@@ -80,6 +84,7 @@ DLL_EXPORT int luaopen_libnnx(lua_State *L)
   nn_DoubleSpatialSparseCriterion_init(L);
   nn_DoubleSpatialMSECriterion_init(L);
   nn_DoubleSpatialClassNLLCriterion_init(L);
+  nn_DoubleSpatialGraph_init(L);
 
   return 1;
 }