fix special View case with -1 arg and dimension set <> 1

added test case

2 files changed, 8 insertions, 1 deletions

diff --git a/View.lua b/View.lua
index 98e53e3..a6785c8 100644
--- a/View.lua
+++ b/View.lua
@@ -30,7 +30,7 @@ local function batchsize(input, size, numInputDims, numElements)
          numElements = 1
          local dim = input:nDimension()
          for i=1,numInputDims do
-            numElements = numElements * input:size(dim-numElements+1)
+            numElements = numElements * input:size(dim-i+1)
          end
       else
          numElements = input:nElement()
diff --git a/test.lua b/test.lua
index 3661d03..f05e502 100644
--- a/test.lua
+++ b/test.lua
@@ -2738,6 +2738,13 @@ function nntest.View()
       minibatch:nElement(),
       "Error in minibatch nElement with size -1")
 
+   -- another setNumInputDims case
+   local minibatch = torch.rand(5,4,10)
+   local module = nn.View(-1):setNumInputDims(2)
+   mytester:assertTableEq(module:forward(minibatch):size(1),
+      minibatch:size(1),
+      "Error in minibatch dimension with size -1")
+   
    -- Minibatch Generalization
    local minibatch = torch.rand(5,2,6)
    local module = nn.View(6)