Answers for "save a torch tensor"

0

save a torch tensor

x = torch.tensor([0, 1, 2, 3, 4])
torch.save(x, 'tensor.pt')
Posted by: Guest on April-27-2022

Python Answers by Framework

Browse Popular Code Answers by Language