Функция PyTorch torch.use_deterministic_algorithms

Pytorch torch 参考手册Справочное руководство Pytorch torch

torch.use_deterministic_algorithmsЭто функция в PyTorch для использования детерминированных алгоритмов. Она определяет, использовать ли детерминированные алгоритмы, когда это возможно.

Определение функции

torch.use_deterministic_algorithms(enabled, warn_only=False)

Пример использования

Пример

import torch

# Включить детерминированные алгоритмы
torch.use_deterministic_algorithms(True)

# Проверить, включено ли это
print(f"Использование детерминированных алгоритмов: {torch.are_deterministic_algorithms_enabled()}")

# Отключить детерминированные алгоритмы
torch.use_deterministic_algorithms(False)
print(f"Использование детерминированных алгоритмов: {torch.are_deterministic_algorithms_enabled()}")

Pytorch torch 参考手册Справочное руководство Pytorch torch

Другие расширения