Merge pull request #20 from sergomezcol/view_module

Add nn.View module

1 files changed, 19 insertions, 0 deletions

diff --git a/test/test.lua b/test/test.lua
index 3c4ec12..5db941a 100644
--- a/test/test.lua
+++ b/test/test.lua
@@ -1867,6 +1867,25 @@ function nntest.SplitTable()
 end
 
 
+function nntest.View()
+   local input = torch.rand(10)
+   local template = torch.rand(5,2)
+   local target = template:size():totable()
+   local module = nn.View(template:size())
+   mytester:assertTableEq(module:forward(input):size():totable(), target, "Error in forward (1)")
+   local module = nn.View(unpack(target))
+   mytester:assertTableEq(module:forward(input):size():totable(), target, "Error in forward (2)")
+
+   -- Minibatch
+   local minibatch = torch.rand(5,10)
+   mytester:assertTableEq(module:forward(minibatch):size(1),
+      minibatch:size(1),
+      "Error in minibatch dimension")
+   mytester:assertTableEq(module:forward(minibatch):nElement(),
+      minibatch:nElement(),
+      "Error in minibatch nElement")
+end
+
 mytester:add(nntest)
 
 if not nn then