PyTorch torch.set_deterministic_algorithm_enabled 函数

PyTorch torch.set_deterministic_algorithm_enabled 函数 Pytorch torch 参考手册 torch.set_deterministic_algorithm_enabled 是 PyTorch 中用于设置确定性算法的函数。设置是否启用确定性算法，启用后每次运行会得到相同的结果。函数定义 torch.set_deterministic_algorithm_enabled(ena..

更新于 2026/7/16

教程详情页默认 noindex。

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()}")

Pytorch torch 参考手册