保存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'])
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值