torch.xpu.manual_seed

torch.xpu.manual_seed(seed) [source]

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

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

Параметры:

seed (int) – Требуемое начальное значение.

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

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