test_torch.py 文件源码

def test_randperm(self):
        _RNGState = torch.get_rng_state()
        res1 = torch.randperm(100)
        res2 = torch.Tensor()
        torch.set_rng_state(_RNGState)
        torch.randperm(res2, 100)
        self.assertEqual(res1, res2, 0)