torch.load() 及 torch.save() 总结_torch.load二进制

1. torch.load()

• 需要用到的一些概念的定义

1. torch: 是pytorch里的一个包，包括了张量(tensor) 的数据结构以及对其的操作

2. pickle 模块: 实现了对一个 Python 对象结构的二进制序列化(pickling)和反序列化(unpickling)。

   pickling: 是一个把Python对象转化成一个字节流的过程。也可称为序列化(serialization)、编组(marshalling)；而相对应反向操作unpickling是将一个字节流转化成一个python对象的过程，也可称为扁平化(flattening)

• torch.load():
  用pickle module读取一个torch.save() 保存的对象(object)，是一个unpicklng的过程

# 参数f: 理解为要加载的文件名; map_location: 可以是函数或者torch.device或者字典，规定存储位置的映射; pickle_module: 前面提到的用来做反序列化的module; **pickle_load_args: optional, 可传入参数到pickle_module.load()中
torch.load
1