diff options
author | Soumith Chintala <soumith@gmail.com> | 2016-07-27 08:18:38 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2016-07-27 08:18:38 +0300 |
commit | 08d0b5db1b56bbb52dfb81a233d288d3654c07d8 (patch) | |
tree | 784391b5541587b125e233702f3be587bcc25e2b | |
parent | 63b4d2d60696965326d8bc226def74deb4c671f9 (diff) |
add one more assertion in test
-rw-r--r-- | test/test_nngraph.lua | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/test/test_nngraph.lua b/test/test_nngraph.lua index a993283..8e5aadc 100644 --- a/test/test_nngraph.lua +++ b/test/test_nngraph.lua @@ -221,6 +221,7 @@ function test.test_gradInputType() tester:eq(torch.typename(module.innode.data.input[1]), "torch.DoubleTensor") tester:eq(torch.typename(module.outnode.data.input[1]), "torch.DoubleTensor") tester:eq(torch.typename(module.forwardnodes[1].data.input[1]), "torch.DoubleTensor") + tester:eq(torch.typename(module.forwardnodes[1].children[1].data.input[1]), "torch.DoubleTensor") module:float() local output = module:forward(input:float()) @@ -230,7 +231,8 @@ function test.test_gradInputType() tester:eq(torch.typename(module.innode.data.input[1]), "torch.FloatTensor") tester:eq(torch.typename(module.outnode.data.input[1]), "torch.FloatTensor") tester:eq(torch.typename(module.forwardnodes[1].data.input[1]), "torch.FloatTensor") - end + tester:eq(torch.typename(module.forwardnodes[1].children[1].data.input[1]), "torch.FloatTensor") + end function test.test_nestedGradInput() local x = nn.Identity()() |