Генератор

class torch.Generator(device='cpu')

Создаёт и возвращает объект генератора, который управляет состоянием алгоритма, производящего псевдослучайные числа. Используется в качестве ключевого аргумента во многих функциях неизменяющей случайной выборки.

Параметры:

device (torch.device, необязательно) – желаемое устройство для генератора.

Возвращает:

Объект torch.Generator.

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

Генератор

Пример:

>>> g_cpu = torch.Generator()
>>> g_cuda = torch.Generator(device='cuda')

device

Generator.device -> device

Получает текущее устройство генератора.

Пример:

>>> g_cpu = torch.Generator()
>>> g_cpu.device
device(type='cpu')

get_state() → Tensor

Возвращает состояние генератора как torch.ByteTensor.

Возвращает:

torch.ByteTensor, содержащий все необходимые биты для восстановления генератора до определённой точки во времени.

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

Тензор

Пример:

>>> g_cpu = torch.Generator()
>>> g_cpu.get_state()

initial_seed() → int

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

Пример:

>>> g_cpu = torch.Generator()
>>> g_cpu.initial_seed()
2147483647

manual_seed(seed) → Generator

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

Параметры:

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

Возвращает:

Объект torch.Generator.

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

Генератор

Пример:

>>> g_cpu = torch.Generator()
>>> g_cpu.manual_seed(2147483647)

seed() → int

Получает недетерминированное случайное число из std::random_device или текущего времени и использует его для инициализации генератора.

Пример:

>>> g_cpu = torch.Generator()
>>> g_cpu.seed()
1516516984916

set_state(new_state) → void

Устанавливает состояние генератора.

Параметры:

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

Пример:

>>> g_cpu = torch.Generator()
>>> g_cpu_other = torch.Generator()
>>> g_cpu.set_state(g_cpu_other.get_state())