torch.manual_seed

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

Задаёт начальное значение для генерации случайных чисел на всех устройствах. Возвращает объект torch.Generator.

Параметры:

seed (int) – Требуемое начальное значение. Значение должно находиться в пределах включительного диапазона [-0x8000_0000_0000_0000, 0xffff_ffff_ffff_ffff]. В противном случае возникает RuntimeError. Отрицательные значения преобразуются в положительные по формуле 0xffff_ffff_ffff_ffff + seed.

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

Generator