Merge pull request #285 from sergomezcol/master

Add support for negative indices in nn.SplitTable

1 files changed, 7 insertions, 0 deletions

diff --git a/test.lua b/test.lua
index 55818e1..82aefc8 100644
--- a/test.lua
+++ b/test.lua
@@ -2929,6 +2929,13 @@ function nntest.SplitTable()
       module = nn.SplitTable(d, 2)
       mytester:asserteq(#module:forward(input), input:size(d+1), "dimension " .. d)
    end
+
+   -- Negative indices
+   local module = nn.SplitTable(-3)
+   local input = torch.randn(3,4,5)
+   mytester:asserteq(#module:forward(input), 3, "negative index")
+   local input = torch.randn(2,3,4,5)
+   mytester:asserteq(#module:forward(input), 3, "negative index (minibatch)")
 end
 
 function nntest.SelectTable()