Add nn.Sequential.remove([index])

1 files changed, 11 insertions, 1 deletions

diff --git a/Sequential.lua b/Sequential.lua
index b08f7df..359a764 100644
--- a/Sequential.lua
+++ b/Sequential.lua
@@ -15,7 +15,7 @@ end
 
 function Sequential:insert(module, index)
    index = index or (#self.modules + 1)
-   if index > (#self.modules + 1) then
+   if index > (#self.modules + 1) or index < 1 then
       error"index should be contiguous to existing modules"
    end
    table.insert(self.modules, index, module)
@@ -23,6 +23,16 @@ function Sequential:insert(module, index)
    self.gradInput = self.modules[1].gradInput
 end
 
+function Sequential:remove(index)
+   index = index or #self.modules
+   if index > #self.modules or index < 1 then
+      error"index out of range"
+   end
+   table.remove(self.modules, index)
+   self.output = self.modules[#self.modules].output
+   self.gradInput = self.modules[1].gradInput
+end
+
 function Sequential:updateOutput(input)
    local currentOutput = input
    for i=1,#self.modules do