脚本之家 服务器常用软件
快捷导航
您的位置：首页 → 脚本专栏 → 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处理图片之PIL模块简单使用方法

    python处理图片之PIL模块简单使用方法

    这篇文章主要介绍了python处理图片之PIL模块简单使用方法,涉及Python使用PIL模块实现针对图片的锐化、绘制直线、绘制椭圆等相关技巧,需要的朋友可以参考下
    2015-05-05
  • python3实现读取chrome浏览器cookie

    python3实现读取chrome浏览器cookie

    这里给大家分享的是python3读取chrome浏览器的cookie（CryptUnprotectData解密）的代码，主要思路是读取到的cookies被封装成字典，可以直接给requests使用。
    2016-06-06
  • Python执行系统命令的五种方式小结

    Python执行系统命令的五种方式小结

    在日常开发中,有时需要在Python脚本中执行系统命令,Python有五种方式来执行系统命令(推荐使用第五种),本文为大家整理了这五种方法的具体使用,希望对大家有所帮助
    2024-01-01
  • pandas将Series转成DataFrame的实现

    pandas将Series转成DataFrame的实现

    本文主要介绍了pandas将Series转成DataFrame的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2023-01-01
  • Tensorflow 多线程设置方式

    Tensorflow 多线程设置方式

    今天小编就为大家分享一篇Tensorflow 多线程设置方式，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
    2020-02-02
  • Python OpenCV 图像区域轮廓标记(框选各种小纸条)

    Python OpenCV 图像区域轮廓标记(框选各种小纸条)

    这篇文章主要介绍了Python OpenCV 图像区域轮廓标记(框选各种小纸条)，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
    2021-03-03
  • Python3中多线程编程的队列运作示例

    Python3中多线程编程的队列运作示例

    这篇文章主要介绍了Python3中多线程编程的队列运作示例,文中用一个简单的例子展示了Python下尝试多线程时队列的进站出站是如何运作的,需要的朋友可以参考下
    2015-04-04
  • Python实现学生管理系统(面向对象版)

    Python实现学生管理系统(面向对象版)

    这篇文章主要为大家详细介绍了Python实现面向对象版的学生管理系统，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
    2021-06-06
  • Python2.x版本中cmp()方法的使用教程

    Python2.x版本中cmp()方法的使用教程

    这篇文章主要介绍了Python2.x版本中cmp()方法的使用教程,在Python3.x版本中该方法不再被内置,需要的朋友可以参考下
    2015-05-05
  • Python机器视觉之基于OpenCV的手势检测

    Python机器视觉之基于OpenCV的手势检测

    这篇文章主要为大家介绍了一个机器视觉项目：基于OpenCV的手势检测，文中的示例代码讲解详细，对我们学习Python和OpenCV有一定的帮助，感兴趣的可以跟随小编学习一下
    2021-12-12

最新评论