Función PyTorch torch.set_flush_denormal

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.set_flush_denormalEs una función en PyTorch para configurar el vaciado de números subnormales. Configurar si vaciar los números subnormales a cero puede mejorar el rendimiento, pero reducirá la precisión.

Definición de la función

torch.set_flush_denormal(flush)

Ejemplo de uso

Ejemplo

import torch

# Habilitar el vaciado de números subnormales
torch.set_flush_denormal(True)
print("Flush denormal enabled")

# Deshabilitar el vaciado de números subnormales
torch.set_flush_denormal(False)
print("Flush denormal disabled")

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones