Pytorch——读写Tensor、读写模型

读写Tensor

import torch 
import torch.nn as nn

存储一个Tensor变量

x = torch.ones(3)
torch.save(x, 'x.pt')
x2 = torch.load('x.pt')
print(x2)

tensor([1., 1., 1.])

存储一个Tensor列表

x = torch.ones(3)
y = torch.zeros(4)
torch.save(