PyTorch torch.set_deterministic_algorithm_enabled 函数
Pytorch torch 参考手册torch.set_deterministic_algorithm_enabled 是 PyTorch 中用于设置确定性算法的函数。设置是否启用确定性算法，启用后每次运行会得到相同的结果。
函数定义
torch.set_deterministic_algorithm_enabled(enabled, mode='warn')
使用示例
实例
import torch
# 设置启用确定性算法
torch.set_deterministic_algorithm_enabled(True)
# 检查是否启用
print(f"Is deterministic: {torch.are_deterministic_algorithms_enabled()}")
# 禁用确定性算法
torch.set_deterministic_algorithm_enabled(False)
print(f"Is deterministic: {torch.are_deterministic_algorithms_enabled()}")
# 设置启用确定性算法
torch.set_deterministic_algorithm_enabled(True)
# 检查是否启用
print(f"Is deterministic: {torch.are_deterministic_algorithms_enabled()}")
# 禁用确定性算法
torch.set_deterministic_algorithm_enabled(False)
print(f"Is deterministic: {torch.are_deterministic_algorithms_enabled()}")