Fonction torch.load de PyTorch
Manuel de référence PyTorch torch
torch.loadest une fonction de PyTorch utilisée pour le chargement. Charge des objets sérialisés à partir d'un fichier.
Définition de fonction
torch.load(f, map_location=None, weights_only=False)
Exemples d'utilisation
Exemple
import torch
# Enregistrer un tenseur
x = torch.tensor([1, 2, 3, 4, 5])
torch.save(x, 'tensor.pt')
# Charger un tenseur
loaded_x = torch.load('tensor.pt')
print(f"Tenseur chargé : {loaded_x}")
# Enregistrer un tenseur
x = torch.tensor([1, 2, 3, 4, 5])
torch.save(x, 'tensor.pt')
# Charger un tenseur
loaded_x = torch.load('tensor.pt')
print(f"Tenseur chargé : {loaded_x}")
Autres extensions