torch.accelerator.random.get_rng_state

torch.accelerator.random.get_rng_state(device=None, /) [исходный код]

Возвращает состояние RNG генератора по умолчанию torch.Generator для текущего ускорителя в виде torch.Tensor типа torch.uint8 для указанного устройства-ускорителя.

Параметры:

device (torch.device, str, int, optional) – Устройство, состояние RNG которого нужно вернуть. Если не указано, по умолчанию используется torch.accelerator.current_device_index().

Возвращает:

состояние RNG генератора по умолчанию для указанного устройства.

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

torch.Tensor

Предупреждение

Эта функция немедленно инициализирует среду выполнения ускорителя.