PyTorch эквивалент numpy np.random.RandomState

Question

Я ищу способ создания случайных объектов без фактического изменения глобального начального числа pytorch.

то есть эквивалент numpy:

rand_gen = np.random.RandomState(seed)
rand_gen.randint(0, 256, self.image_dim))  # for example

Answer 1

Вы можете передать ваш torch.Generator вручную случайной функции. Я думаю, что этот код должен работать:

gen0 = torch.Generator()
gen1 = torch.Generator()

gen0 = gen0.manual_seed(0)
gen1 = gen1.manual_seed(1)
torch.rand(5, generator=gen0)
torch.rand(5, generator=gen0)
torch.rand(5, generator=gen1)
torch.rand(5, generator=gen1)

gen0 = gen0.manual_seed(0)
gen1 = gen1.manual_seed(1)
torch.rand(5, generator=gen1)
torch.rand(5, generator=gen1)
torch.rand(5, generator=gen0)
torch.rand(5, generator=gen0)