PyTorch torch.set_flush_denormal 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.set_flush_denormalPyTorch で非正規数をフラッシュするための関数です。非正規数をゼロにフラッシュするかどうかを設定し、パフォーマンスを向上させることができますが、精度は低下します。

関数定義

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 参考手册Pytorch torch リファレンスマニュアル

その他の拡張