Guardar un tensor de antorcha

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