Pytorch read / write file

Writing documents

torch.save()

import torch 
from torch import nn 
from torch.nn import functional as F

x= torch.arange(4)
torch.save(x, "x-file") #  The program will automatically x Write to x-file In file , If you start x-file File does not exist , The system will automatically create the file name 

Reading documents

torch.load()

import torch 
from torch import nn 
from torch.nn import functional as F

x= torch.arange(4)
torch.save(x, "x-file")
print(x)

x2 = torch.load("x-file") # Read the file 
print(x2)