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")
# 启用刷新非正规数
torch.set_flush_denormal(True)
print("Flush denormal enabled")
# 禁用刷新非正规数
torch.set_flush_denormal(False)
print("Flush denormal disabled")