PyTorch torch.set_flush_denormal 函数

PyTorch torch.set_flush_denormal 函数 Pytorch torch 参考手册 torch.set_flush_denormal 是 PyTorch 中用于设置刷新非正规数的函数。设置是否刷新非正规数到零，可以提高性能但会降低精度。函数定义 torch.set_flush_denormal(flush) 使用示例实例 [mycode4 type='python'] imp..

更新于 2026/7/16

教程详情页默认 noindex。

PyTorch torch.set_flush_denormal 函数

Pytorch torch 参考手册

torch.set_flush_denormal 是 PyTorch 中用于设置刷新非正规数的函数。设置是否刷新非正规数到零，可以提高性能但会降低精度。

函数定义

torch.set_flush_denormal(flush)

使用示例

实例

import torch

# 启用刷新非正规数
torch.set_flush_denormal(True)
print("Flush denormal enabled")

# 禁用刷新非正规数
torch.set_flush_denormal(False)
print("Flush denormal disabled")

Pytorch torch 参考手册