Fonction torch._assert de PyTorch
Manuel de référence PyTorch torch
torch._assertC'est une fonction utilisée pour la vérification des assertions dans PyTorch. Si la condition est False, elle lève une RuntimeError.
Définition de fonction
torch._assert(condition, message)
Exemple d'utilisation
Exemple
import torch
# Exemple de code
condition = True
torch._assert(condition, "Échec de l'assertion !")
print("Assertion réussie !")
# Exemple d'échec
try:
torch._assert(False, "Cela va échouer")
except RuntimeError as e:
print(f"Erreur capturée : {e}")
# Exemple de code
condition = True
torch._assert(condition, "Échec de l'assertion !")
print("Assertion réussie !")
# Exemple d'échec
try:
torch._assert(False, "Cela va échouer")
except RuntimeError as e:
print(f"Erreur capturée : {e}")
Autres extensions