Method test_sum

test/unit/test_callify.py:17–20

(self)

    self.assertEqual(out.item(), 5.0)

  def test_sum(self):
    out = Tensor.ones(16).contiguous().sum()
    out.callify()
    self.assertEqual(out.item(), 16.0)

  def test_multi_output(self):
    a = Tensor([1.,2,3])