“Sauver un tenseur de torche” Réponses codées

Sauver un tenseur de torche

x = torch.tensor([0, 1, 2, 3, 4])
torch.save(x, 'tensor.pt')