Fonction torch.set_flush_denormal de PyTorch

Pytorch torch 参考手册Manuel de référence Pytorch torch

torch.set_flush_denormalC'est une fonction de PyTorch pour purger les nombres dénormalisés. Le fait de définir si les nombres dénormalisés doivent être mis à zéro peut améliorer les performances mais réduire la précision.

Définition de fonction

torch.set_flush_denormal(flush)

Exemple d'utilisation

Exemple

import torch

# Activer la purge des nombres dénormalisés
torch.set_flush_denormal(True)
print("Purge des nombres dénormalisés activée")

# Désactiver la purge des nombres dénormalisés
torch.set_flush_denormal(False)
print("Purge des nombres dénormalisés désactivée")

Pytorch torch 参考手册Manuel de référence Pytorch torch

Autres extensions