脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → python → Pytorch tensor序列化和并行化

Pytorch实现tensor序列化和并行化的示例详解

 更新时间：2023年12月08日 11:27:28   作者：怎么开心怎么玩  
这篇文章主要介绍了Pytorch实现tensor序列化和并行化,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,感兴趣的同学们下面随着小编来一起学习学习吧

Pytorch 实现 tensor 序列化和反序列化

序列化 Serialization

保存一个对象到一个硬盘文件上

torch.save(obj, f, pickle_mode, pickle_protocol)

参数：

  • obj – 保存对象
  • f - 类文件对象 (返回文件描述符)或一个保存文件名的字符串
  • pickle_module – 用于 pickling 元数据和对象的模块
  • pickle_protocol – 指定 pickle protocal 可以覆盖默认参数

从磁盘文件中读取一个通过 torch.save()保存的对象

torch.load(f, map_location=None, pickle_mode)

torch.load() 可通过参数 map_location 动态地进行内存重映射，使其能从不动设备中读取文件。一般调用时，需两个参数: storage 和 location tag。 返回不同地址中的 storage，或着返回 None (此时地址可以通过默认方法进行解析)。 如果这个参数是字典的话，意味着其是从文件的地址标记到当前系统的地址标记的映射。 默认情况下，location tags 中 "cpu"对应 host tensors,‘cuda:device_id’ (e.g.‘cuda:2’) 对应 cuda tensors。 用户可以通过 register_package 进行扩展，使用自己定义的标记和反序列化方法。

  • f – 类文件对象 (返回文件描述符)或一个保存文件名的字符串
  • map_location – 一个函数或字典规定如何 remap 存储位置
  • pickle_module – 用于 unpickling 元数据和对象的模块 (必须匹配序列化文件时的 pickle_module )

并行化 Parallelism

torch.get_num_threads
torch.get_num_threads() → int
#获得用于并行化 CPU 操作的 OpenMP 线程数
torch.set_num_threads
torch.set_num_threads(int)
#设定用于并行化 CPU 操作的 OpenMP 线程数

以上就是Pytorch实现tensor序列化和并行化的示例详解的详细内容，更多关于Pytorch tensor序列化和并行化的资料请关注脚本之家其它相关文章！

您可能感兴趣的文章:

相关文章

  • Python内置的HTTP协议服务器SimpleHTTPServer使用指南

    Python内置的HTTP协议服务器SimpleHTTPServer使用指南

    这篇文章主要介绍了Python内置的HTTP协议服务器SimpleHTTPServer使用指南,SimpleHTTPServer本身的功能十分简单,文中介绍了需要的朋友可以参考下
    2016-03-03
  • 详解Pycharm第三方库的安装及使用方法

    详解Pycharm第三方库的安装及使用方法

    这篇文章主要介绍了详解Pycharm第三方库的安装及使用方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2020-12-12
  • Python字符串查找基本操作代码案例

    Python字符串查找基本操作代码案例

    这篇文章主要介绍了Python字符串查找基本操作代码案例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-10-10
  • 聊聊python中not 与 is None的区别

    聊聊python中not 与 is None的区别

    这篇文章主要介绍了在python中not 与 is None的区别，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
    2021-05-05
  • Python 如何提高元组的可读性

    Python 如何提高元组的可读性

    这篇文章主要介绍了Python 如何提高元组的可读性,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2019-08-08
  • 用python实现批量重命名文件的代码

    用python实现批量重命名文件的代码

    任务很简单，某个目录下面有几千个文件，某些文件没有后缀名，现在的任务就是将所有的没有后缀名的文件加上后缀名，python有现成的函数可以实现，但是在实现过程中遇到几个问题，分享一下解决方法
    2012-05-05
  • Python定时任务sched模块用法示例

    Python定时任务sched模块用法示例

    这篇文章主要介绍了Python定时任务sched模块用法,结合实例形式分析了定时任务sched模块的基本功能、参数含义、多线程定时任务等相关操作技巧,需要的朋友可以参考下
    2018-07-07
  • 一篇文章带你学习Python3的高级特性(1)

    一篇文章带你学习Python3的高级特性(1)

    这篇文章主要为大家详细介绍了Python3的高阶函数，主要介绍什么是高级特性，高级特性的用法，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2022-01-01
  • Python爬取微信小程序Charles实现过程图解

    Python爬取微信小程序Charles实现过程图解

    这篇文章主要介绍了Python爬取微信小程序Charles实现过程图解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
    2020-09-09
  • 详解python中Numpy的属性与创建矩阵

    详解python中Numpy的属性与创建矩阵

    这篇文章给大家分享了关于python中Numpy的属性与创建矩阵的相关知识点内容，有兴趣的朋友们可以学习参考下。
    2018-09-09

最新评论