[pytroch] 存储torch到文件

# 存储数据
import scipy.io as sio
output_file = "xx.mat"
output_dict = {"a":data,"b":data}
sio.savemat(output_file,output_dict)

# 读取数据
sio.loadmat(output_file)