From 0063dbdefd77182c9235315d5574caf7bbf25147 Mon Sep 17 00:00:00 2001 From: Ronan Collobert Date: Tue, 21 Apr 2015 22:11:04 -0700 Subject: fix View (hopefuly) now handles better -1 cases --- test.lua | 11 +++++++++++ 1 file changed, 11 insertions(+) (limited to 'test.lua') diff --git a/test.lua b/test.lua index 23c7fbd..359973d 100644 --- a/test.lua +++ b/test.lua @@ -2798,6 +2798,17 @@ function nntest.View() minibatch:size(1), "Error in minibatch dimension with size -1") + -- another setNumInputDims case + local minibatch = torch.rand(2,5,4,10) + local module = nn.View(4,-1):setNumInputDims(2) + local out = module:forward(minibatch) + mytester:assertTableEq(out:size(1), minibatch:size(1)*minibatch:size(2), + "Error in minibatch dimension with size -1") + mytester:assertTableEq(out:size(2), minibatch:size(3), + "Error in minibatch dimension with size -1") + mytester:assertTableEq(out:size(3), minibatch:size(4), + "Error in minibatch dimension with size -1") + -- Minibatch Generalization local minibatch = torch.rand(5,2,6) local module = nn.View(6) -- cgit v1.2.3