torch.accelerator.random.get_rng_state_all

torch.accelerator.random.get_rng_state_all() [source]

Возвращает список torch.Tensor типа torch.uint8, представляющих состояния генератора случайных чисел всех устройств для текущего ускорителя.

Возвращает:

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

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

list[torch.Tensor]

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

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