X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=test-dagnn.lua;h=d7179cc1b5110edb2b6666ce8a1604f6fe6d2102;hb=31dc42fc93ed12491ceb10ef3bfc4296878380ee;hp=0c9fe6d57376b8be5a3d2621677e1bd667f429a9;hpb=941ae69a9ccf274e3bff8b1fa45da04fb825a647;p=dagnn.git

diff --git a/test-dagnn.lua b/test-dagnn.lua
index 0c9fe6d..d7179cc 100755
--- a/test-dagnn.lua
+++ b/test-dagnn.lua
@@ -42,14 +42,14 @@ g:addEdge(b, c)
 g:addEdge(b, d)
 g:addEdge(d, f)
 
-g:setInput({ a })
+g:setInput({{a}})
 g:setOutput({ e, f })
 
 g:print()
 
 input = torch.Tensor(3, 10):uniform()
 
-output = g:updateOutput({ input })
+output = g:updateOutput({{ input }})
 
 printTensorTable(output)
 
@@ -58,6 +58,6 @@ printTensorTable(output)
 print('******************************************************************')
 print('** updateGradInput ***********************************************')
 print('******************************************************************')
-gradInput = g:updateGradInput({ input }, output)
+gradInput = g:updateGradInput({{input}}, output)
 
 printTensorTable(gradInput)