torch.set_flush_denormal

torch.set_flush_denormal(mode) → bool

Отключает денормализованные числа с плавающей точкой на CPU.

Возвращает True, если ваша система поддерживает сброс денормализованных чисел и ей удалось настроить режим сброса денормализованных чисел. set_flush_denormal() поддерживается на архитектурах x86 с поддержкой SSE3 и архитектуре AArch64.

Параметры:

mode (bool) – Определяет, включать ли режим сброса денормализованных чисел

Пример:

>>> torch.set_flush_denormal(True)
True
>>> torch.tensor([1e-323], dtype=torch.float64)
tensor([ 0.], dtype=torch.float64)
>>> torch.set_flush_denormal(False)
True
>>> torch.tensor([1e-323], dtype=torch.float64)
tensor(9.88131e-324 *
       [ 1.0000], dtype=torch.float64)