torch.mtia.seed

torch.mtia.seed() [source]

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

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

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

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

int