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

added test case

1 files changed, 1 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()