From ddfe535a7900939136f40773847f5b1152643a2a Mon Sep 17 00:00:00 2001 From: Sergio Gomez Date: Mon, 30 Jun 2014 16:39:16 +0100 Subject: Add setNumInputDims() method to nn.View This allows to use minibatches when using -1 for one of the sizes. --- test/test.lua | 7 +++++++ 1 file changed, 7 insertions(+) (limited to 'test') diff --git a/test/test.lua b/test/test.lua index 5e4bce7..88adf96 100644 --- a/test/test.lua +++ b/test/test.lua @@ -1884,6 +1884,13 @@ function nntest.View() mytester:assertTableEq(module:forward(minibatch):nElement(), minibatch:nElement(), "Error in minibatch nElement") + local module = nn.View(-1):setNumInputDims(1) + mytester:assertTableEq(module:forward(minibatch):size(1), + minibatch:size(1), + "Error in minibatch dimension with size -1") + mytester:assertTableEq(module:forward(minibatch):nElement(), + minibatch:nElement(), + "Error in minibatch nElement with size -1") end -- Define a test for SpatialUpSamplingCuda -- cgit v1.2.3