То же самое случайное семя для Numpy и PyTorch

Я пытаюсь сравнить свои функции в ванильном Python/Numpy с функциями PyTorch. Для этого я хотел бы установить семена обоих, чтобы сделать результаты сопоставимыми.

Можно ли установить семена для NumPy, как np.random.seed(x) и PyTorch как torch.manual_seed(x) так что оба производят одинаковые случайные числа?

Щас с таким же x значение np.random.uniform(0, 1) а также torch.rand(1) производить разные выходы.

1 ответ

Что вы думаете об этом решении?

Вместо

      np.random.seed(0)

np.random.rand(0, 1)

ты мог бы использовать

      torch.manual_seed(0) 

torch.rand(1).numpy()

Таким образом, у вас будут одинаковые случайные числа между PyTorch и NumPy, но все равно как массив NumPy, если хотите...

Другие вопросы по тегам