Add MaskedSelect.lua module to replicate torch.maskedSelect behaviour.

1 files changed, 1 insertions, 0 deletions

diff --git a/init.lua b/init.lua
index 0a1a0c8..5d64cfb 100644
--- a/init.lua
+++ b/init.lua
@@ -33,6 +33,7 @@ require('nn.Transpose')
 require('nn.BatchNormalization')
 require('nn.Padding')
 require('nn.GradientReversal')
+require('nn.MaskedSelect')
 
 require('nn.Copy')
 require('nn.Min')