Есть ли в PyTorch объект, подобный RandomState, для генерации случайных чисел?

Question

в numpy я могу

import numpy as np

rs = np.random.RandomState(seed=0)

и затем передать этот объект, например, для внедрения зависимости.

Имеет ли PyTorch похожий интерфейс?Я ничего не могу найти в документах, но, может быть, я что-то упустил.

Answer 1

Ближайшей вещью будет torch.manual_seed, которая устанавливает начальное значение для генерации случайных чисел и возвращает torch.Generator.Этот поток здесь содержит больше информации, по-видимому, могут быть некоторые несоответствия в зависимости от того, используете ли вы графический процессор или процессор.