def test_cumsum(self): x = torch.rand(100, 100) res1 = torch.cumsum(x, 1) res2 = torch.Tensor() torch.cumsum(res2, x, 1) self.assertEqual(res1, res2)