torch.xpu.seed

torch.xpu.seed() [исходный код]

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

Эту функцию можно безопасно вызывать, если XPU недоступен; в таком случае вызов будет проигнорирован без уведомления.

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

Если вы работаете с моделью на нескольких GPU, эта функция инициализирует начальное значение только на одном GPU. Чтобы инициализировать все GPU, используйте seed_all().