Added SpatialMaxSampling module, for flexible competitive resampling.

1 files changed, 5 insertions, 0 deletions

diff --git a/init.c b/init.c
index 764a18c..50ac755 100644
--- a/init.c
+++ b/init.c
@@ -24,6 +24,9 @@ static const void* torch_DoubleTensor_id = NULL;
 #include "generic/SpatialReSampling.c"
 #include "THGenerateFloatTypes.h"
 
+#include "generic/SpatialMaxSampling.c"
+#include "THGenerateFloatTypes.h"
+
 #include "generic/SparseCriterion.c"
 #include "THGenerateFloatTypes.h"
 
@@ -59,6 +62,7 @@ DLL_EXPORT int luaopen_libnnx(lua_State *L)
   nn_FloatThreshold_init(L);
   nn_FloatSpatialUpSampling_init(L);
   nn_FloatSpatialReSampling_init(L);
+  nn_FloatSpatialMaxSampling_init(L);
   nn_FloatDistMarginCriterion_init(L);
   nn_FloatSparseCriterion_init(L);
   nn_FloatSpatialSparseCriterion_init(L);
@@ -73,6 +77,7 @@ DLL_EXPORT int luaopen_libnnx(lua_State *L)
   nn_DoubleThreshold_init(L);
   nn_DoubleSpatialUpSampling_init(L);
   nn_DoubleSpatialReSampling_init(L);
+  nn_DoubleSpatialMaxSampling_init(L);
   nn_DoubleDistMarginCriterion_init(L);
   nn_DoubleSparseCriterion_init(L);
   nn_DoubleSpatialSparseCriterion_init(L);