Функция PyTorch torch.set_flush_denormal

Pytorch torch 参考手册 Справочное руководство Pytorch torch

torch.set_flush_denormalЭто функция в PyTorch для сброса денормализованных чисел в ноль. Установка того, сбрасывать ли денормализованные числа в ноль, может повысить производительность, но снизить точность.

Определение функции

torch.set_flush_denormal(flush)

Пример использования

Пример

import torch

# Включить сброс денормализованных чисел
torch.set_flush_denormal(True)
print("Сброс денормализованных чисел включён")

# Отключить сброс денормализованных чисел
torch.set_flush_denormal(False)
print("Сброс денормализованных чисел отключён")

Pytorch torch 参考手册 Справочное руководство Pytorch torch

Другие расширения