保存tensor至本地文件

由于实验过程需要,将中间结果tensor保存至本地文件,这里利用numpy库中的savez函数来实现,具体的代码如下

# 首先将要保存的tensor从计算图中分离出来,这里用到detach函数
enc = by_latent.detach().cpu().numpy()

因为用的是savez函数来实现保存,所以直接拿来用就可以,保存的默认文件后缀是*.npz

np.savez("/test/enc_{}".format(batch_idx), codecs)
# np.savez_compressed("/test/enc_{}".format(batch_idx), codecs)  # 也可以用进行压缩存储

重新加载保存的npz文件

enc = np.load(path)
# 默认的索引是arr_0,后面以此类推
print(enc['arr_0'])

你可能感兴趣的:(Torch)