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