torch.accelerator.random.initial_seed

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

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

Параметры:

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

Возвращает:

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

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

int

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

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