test_torch.py 文件源码

def test_randn(self):
        torch.manual_seed(123456)
        res1 = torch.randn(SIZE, SIZE)
        res2 = torch.Tensor()
        torch.manual_seed(123456)
        torch.randn(res2, SIZE, SIZE)
        self.assertEqual(res1, res2)