在磁盘中保存任何内容的最简单方法是使用pickle：

import pickle
import torch

a = torch.rand(3,4,5)

# save
with open('filename.pickle', 'wb') as handle:
    pickle.dump(a, handle)

# open
with open('filename.pickle', 'rb') as handle:
    b = pickle.load(handle)

字符串
你也可以直接用pytorch保存东西，但那只是pikle的一个pytorch Package 。

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

型
如果你想在一个文件中保存多个Tensor，你可以将它们 Package 在一个字典中：

import torch
x = torch.tensor([0, 1, 2, 3, 4])
a = torch.rand(2,3,4,5)
b = torch.zeros(37)
torch.save({"a": a, "b":b, "x", x}, 'tensors.pt')

型

h5py允许你将大量Tensor保存到同一个文件中，你不必将整个文件的内容放入内存。h5py将直接将Tensor存储到磁盘，您可以在需要时加载Tensor。它允许在加载和保存时对这些Tensor进行切片，其以类似的方式工作，即不需要将整个Tensor加载到内存中，以便加载它的一个切片，或者为了保存它的一个切片。