diff options
author | Soumith Chintala <soumith@gmail.com> | 2015-07-31 19:35:25 +0300 |
---|---|---|
committer | Soumith Chintala <soumith@gmail.com> | 2015-07-31 19:35:25 +0300 |
commit | 75a82027d3817256e4aac0077006bd3c380749f0 (patch) | |
tree | aed6658032628efcabed09654be04b28b24cc945 | |
parent | a648065d8eb0fcafba97c0b679aeeb00ad048e5e (diff) | |
parent | 28f2072373fe90fe785fd4ea204f2deff3a521e6 (diff) |
Merge pull request #340 from dominikgrewe/singleton_fix
Update nntest.addSingletonDimension to test adding final dimension.
-rw-r--r-- | test.lua | 2 |
1 files changed, 1 insertions, 1 deletions
@@ -4023,7 +4023,7 @@ function nntest.addSingletonDimension() mytester:assertTensorEq(firstDim[1], tensor, 0, "wrong content for singleton dimension 1") - local dim = torch.random(dims) + local dim = torch.random(dims + 1) local result = nn.utils.addSingletonDimension(tensor, dim) local resultSize = size:totable() table.insert(resultSize, dim, 1) |