fix assertTensorEq for case of empty tensors

The case of empty tensors was handled wrongly: leading to exception. In the test of torch.maskedSelect and similar the case of an empty tensor can occur, leading to occasional test failures.

1 files changed, 3 insertions, 0 deletions

diff --git a/Tester.lua b/Tester.lua
index 42aef73..ebbe8a9 100644
--- a/Tester.lua
+++ b/Tester.lua
@@ -53,6 +53,9 @@ function Tester:assertne (val, condition, message)
 end
 
 function Tester:assertTensorEq(ta, tb, condition, message)
+   if ta:dim() == 0 and tb:dim() == 0 then
+      return
+   end
    local diff = ta-tb
    local err = diff:abs():max()
    self:assert_sub(err<condition,string.format('%s\n%s  val=%s, condition=%s',(message or ''),' TensorEQ(==) violation ', tostring(err), tostring(condition)))