Search
 
SCRIPT & CODE EXAMPLE
 
CODE EXAMPLE FOR PYTHON

save a torch tensor

x = torch.tensor([0, 1, 2, 3, 4])
torch.save(x, 'tensor.pt')
Source by pytorch.org #
 
PREVIOUS NEXT
Tagged: #save #torch #tensor
ADD COMMENT
Topic
Name
1+9 =