diff options
-rw-r--r-- | View.lua | 2 | ||||
-rw-r--r-- | test.lua | 7 |
2 files changed, 8 insertions, 1 deletions
@@ -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() @@ -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) |