由于实验过程需要,将中间结果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'])