码迷,mamicode.com
首页 > 其他好文 > 详细

2.7 序列化(保存)tensor

时间:2020-07-30 14:02:41      阅读:67      评论:0      收藏:0      [点我收藏+]

标签:inf   rds   tensor   序列化   通用   info   nump   分数   mamicode   

可以直接调用tensor的save和load方法对tensor进行保存和读取.

保存:

技术图片

 

 读取:

技术图片

 

 或者

技术图片

 

 但是这样保存下来的只能被Pytorch所识别。可以利用 h5py 库进行通用格式的保存。

首先要转换成numpy数组,然后调用  create_dataset 函数,保存为hdf5格式的文件

技术图片

 

 这里的 coords是我们定义的key值。hdf5格式的文件是基于key-value保存的,所以我们在保存的时候要指定一个key。

hdf5文件支持从硬盘上部分读取,比如我们只读取coords的部分数据

技术图片

当文件指针被打开,或者数据集被加载时,pytor不会马上加载数据,之后当我们需要取具体的数据时,才会从硬盘上读取我们

所需要的数据。读取到的数据可以看做是一个numpy数组。因此我们可以直接把读取到的数据转换成tensor。

技术图片

 

 最后,记得关闭文件。

 

2.7 序列化(保存)tensor

标签:inf   rds   tensor   序列化   通用   info   nump   分数   mamicode   

原文地址:https://www.cnblogs.com/superxuezhazha/p/13402987.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!