add tests

Roy Li · Roy Li · commit 229472b1081d · 2018-06-06T19:00:58.000-07:00
diff --git a/test/test_nn.py b/test/test_nn.py
@@ -4116,6 +4116,16 @@ def func(root):
         gradcheck(func, [v])
         gradgradcheck(func, [v])
 
+    def test_bce_loss_always_nonnegative(self):
+        target = torch.ones(5)
+        input = torch.ones(5)
+        assert ((nn.BCELoss()(input,target) < 0).sum() == 0)
+
+        target = torch.zeros(5)
+        input = torch.zeros(5)
+        print((nn.BCELoss()(input,target) < 0).sum)
+        assert ((nn.BCELoss()(input,target) < 0).sum() == 0)
+
     def test_bce_with_logits_raises_if_target_and_input_are_different_size(self):
         target = torch.rand(5)
         input = torch.rand(5, 1)
