Add SpatialDownSampling module, a simple down-sampling averaging the pixels

1 files changed, 5 insertions, 0 deletions

diff --git a/init.c b/init.c
index efc5081..2b5afcc 100644
--- a/init.c
+++ b/init.c
@@ -21,6 +21,9 @@ static const void* torch_DoubleTensor_id = NULL;
 #include "generic/SpatialUpSampling.c"
 #include "THGenerateFloatTypes.h"
 
+#include "generic/SpatialDownSampling.c"
+#include "THGenerateFloatTypes.h"
+
 #include "generic/SpatialReSampling.c"
 #include "THGenerateFloatTypes.h"
 
@@ -64,6 +67,7 @@ DLL_EXPORT int luaopen_libnnx(lua_State *L)
   nn_FloatAbs_init(L);
   nn_FloatThreshold_init(L);
   nn_FloatSpatialUpSampling_init(L);
+  nn_FloatSpatialDownSampling_init(L);
   nn_FloatSpatialReSampling_init(L);
   nn_FloatSpatialMaxSampling_init(L);
   nn_FloatDistMarginCriterion_init(L);
@@ -80,6 +84,7 @@ DLL_EXPORT int luaopen_libnnx(lua_State *L)
   nn_DoubleAbs_init(L);
   nn_DoubleThreshold_init(L);
   nn_DoubleSpatialUpSampling_init(L);
+  nn_DoubleSpatialDownSampling_init(L);
   nn_DoubleSpatialReSampling_init(L);
   nn_DoubleSpatialMaxSampling_init(L);
   nn_DoubleDistMarginCriterion_init(L);