摘要:
可以直接调用tensor的save和load方法对tensor进行保存和读取. 保存: 读取: 或者 但是这样保存下来的只能被Pytorch所识别。可以利用 h5py 库进行通用格式的保存。 首先要转换成numpy数组,然后调用 create_dataset 函数,保存为hdf5格式的文件 这里的 阅读全文
摘要:
pytorh的tensor与numpy的转换,是通过Python buffer protocol完成的。所以这个转换时 zero-copy的。 tensor 》numpy 转换之后的numpy数组,与tensor共用一个基础buffer,所以如果tensor是存储在CPU上的,他们对数据的修改会相互 阅读全文