Equivalent of random.seed() in pymanopt manifold

[Beñat Mencia]

I want to seed a pseudo-random number generator in the Stiefel manifold. Usually I would do:

import random

random.seed(9001)   

random.randint(1, 10) 

Now, I want to do the equivalent in the Stiefel manifold. I tried:

import pymanopt.manifolds

import random

manifold = pymanopt.manifolds.Stiefel(4,2, k=1)

random.seed(9002)

x0 = manifold.rand()

print(x0)

But this does not generate the same x0 every time. 

Any ideas?

[Nicolas Boumal]

Hello Beñat,

Looking into the code of Stiefel.rand(), the random numbers are generated using:

np.random.randn(self._n, self._p)

where np is NumPy. I recommend you look into the documentation of the random number generator of NumPy, or here for example:

https://stackoverflow.com/questions/21494489/what-does-numpy-random-seed0-do

Best,

Nicolas