Python Tensor.sum Examples

Example #1

File: test_ops.py Project: shivral/tinygrad

 def test_sum(self):
   helper_test_op([(45,3)], lambda x: x.sum(), Tensor.sum)
   helper_test_op([(3,4,5,6)], lambda x: x.sum(axis=3), lambda x: Tensor.sum(x, axis=3))
   helper_test_op([(3,4,5,6)], lambda x: x.sum(axis=(1,3)), lambda x: Tensor.sum(x, axis=(1,3)))
   helper_test_op([(3,4,5,6)], lambda x: x.sum(axis=(0,2)), lambda x: Tensor.sum(x, axis=(0,2)))
   helper_test_op([(3,4,5,6)], lambda x: x.sum(axis=(1,2)), lambda x: Tensor.sum(x, axis=(1,2)))
   helper_test_op([(3,4,5,6)], lambda x: x.sum(axis=1), lambda x: Tensor.sum(x, axis=1))

Example #2

File: test_ops.py Project: vincent9029/tinygrad

 def test_sum(self):
     helper_test_op([(45, 3)],
                    lambda x: x.sum(),
                    Tensor.sum,
                    device=self.device)
     helper_test_op([(3, 4, 5, 6)],
                    lambda x: x.sum(axis=(1, 2)),
                    lambda x: Tensor.sum(x, axis=(1, 2)),
                    device=self.device)
     helper_test_op([(3, 4, 5, 6)],
                    lambda x: x.sum(axis=1),
                    lambda x: Tensor.sum(x, axis=1),
                    device=self.device)

Example #3

File: test_ops.py Project: shezhanjun/tinygrad

 def test_sum_axis(self):
     helper_test_op([(3, 4, 5, 6)],
                    lambda x: x.sum(axis=(1, 2)),
                    lambda x: Tensor.sum(x, axis=(1, 2)),
                    device=self.device)