torch.random

torch.random.fork_rng(devices=None, enabled=True, _caller='fork_rng', _devices_kw='devices', device_type='cuda') [source]

Разветвляет генератор случайных чисел, так что при возврате генератор случайных чисел восстанавливается в предыдущем состоянии.

Параметры

• devices (итерируемый список идентификаторов устройств) – устройства, для которых необходимо разветвить генератор случайных чисел. Состояние генератора случайных чисел CPU всегда разветвляется. По умолчанию, fork_rng() работает со всеми устройствами, но выведет предупреждение, если у вашей машины много устройств, так как эта функция будет работать очень медленно в этом случае. Если вы явно укажете устройства, это предупреждение будет подавлено
• enabled (bool) – если False, генератор случайных чисел не разветвляется. Это удобный аргумент для лёгкого отключения контекстного менеджера, без необходимости удалять его и отступать код Python под ним.
• deivce_type (str) – тип устройства str, по умолчанию cuda. Что касается пользовательского устройства, см. подробности в [Примечание: поддержка пользовательского устройства с privateuse1]

Тип возвращаемого значения

Генератор

torch.random.get_rng_state() [source]

Возвращает состояние генератора случайных чисел в виде torch.ByteTensor.

Тип возвращаемого значения

Тензор

torch.random.initial_seed() [source]

Возвращает начальное значение для генерации случайных чисел в виде Python long.

Тип возвращаемого значения

int

torch.random.manual_seed(seed) [source]

Устанавливает начальное значение для генерации случайных чисел. Возвращает torch.Generator объект.

Параметры

seed (int) – Желаемое значение начального значения. Значение должно находиться в диапазоне [-0x8000_0000_0000_0000, 0xffff_ffff_ffff_ffff]. В противном случае, возникает исключение RuntimeError. Отрицательные входные данные пересчитываются в положительные значения по формуле 0xffff_ffff_ffff_ffff + seed.

Тип возвращаемого значения

Генератор

torch.random.seed() [source]

Устанавливает начальное значение для генерации случайных чисел на не детерминированное случайное число. Возвращает 64-битное число, используемое для инициализации генератора случайных чисел.

Тип возвращаемого значения

int

torch.random.set_rng_state(new_state) [source]

Устанавливает состояние генератора случайных чисел.

Параметры

new_state (torch.ByteTensor) – Желаемое состояние