首页 > 行业资讯 > PyTorch 教程-PyTorch 包

PyTorch 教程-PyTorch 包

时间:2023-12-18 来源: 浏览:

PyTorch 教程-PyTorch 包

点击关注 Python架构师
Python架构师

gh_1d7504e4dee1

回复:python,领取Python面试题。分享Python教程,Python架构师教程,Python爬虫,Python编程视频,Python脚本,Pycharm教程,Python微服务架构,Python分布式架构,Pycharm注册码。

整理: python架构师

PyTorch是一个针对使用CPU和GPU进行深度学习的优化张量库。PyTorch拥有丰富的包,用于执行深度学习概念。这些包帮助我们进行优化、转换、损失计算等。让我们简要了解这些包。

序号 名称 描述
1. Torch torch包包含用于多维张量的数据结构,以及对这些张量的数学运算的定义。
2. torch.Tensor 该包是一个包含单一数据类型元素的多维矩阵。
3. 张量属性
a) torch.dtype 表示torch.Tensor数据类型的对象。
b) torch.device 表示将分配torch.Tensor的设备的对象。
c) torch.layout 表示torch.Tensor的内存布局的对象。
4. 类型信息 可以通过torch.iinfo或torch.finfo访问torch.dtype的数值属性。
1) torch.finfo 表示浮点torch.dtype的数值属性的对象。
2) torch.iinfo 表示整数torch.dtype的数值属性的对象。
5. torch.sparse Torch支持以COO(坐标)格式存储和处理稀疏张量,这将高效地存储和处理大多数元素为零的张量。
6. torch.cuda Torch支持CUDA张量类型,其实现与CPU张量相同,但在计算方面利用了GPU。
7. torch.Storage torch.Storage是一个连续的、一维的单一数据类型数组。
8. torch.nn 该包为我们提供了许多用于实现和训练神经网络的类和模块。
9. torch.nn.functional 该包具有与torch.nn类似的功能类。
10. torch.optim 该包用于实现各种优化算法。
11. torch.autograd 该包提供了用于实现任意标量值函数的自动微分的类和函数。
12. torch.distributed 该包支持三种不同功能的后端。
13. torch.distribution 该包允许我们构建随机计算图,并为优化提供随机梯度估计器。
14. torch.hub 它是一个预训练模型仓库,旨在促进研究的可重现性。
15. torch.multiprocessing 它是对本机multiprocessing模块的包装。
16. torch.utils.bottleneck 它是一个用于在程序中调试瓶颈的初始步骤的工具。
17. torch.utils.checkpoint 用于在源程序中创建检查点。
18. torch.tils.cpp_extension 用于创建C++、CUDA和其他语言的扩展。
19. torch.utils.data 该包主要用于创建数据集。
20. torch.utils.dlpack 用于将Dlpack解码为张量。
21. torch.onnx ONNX导出器是一种基于跟踪的导出器,这意味着它通过执行您的模型一次并导出实际在此运行期间运行的运算符来操作。

资源分享

点击领取:最全Python资料合集
参考:
https://pytorch.org/docs/stable/index.html

 
热门推荐
  • Python爬取抖音短视频(无水印版)
  • PyTorch 教程-什么是PyTorch?
  • 如何使用并订阅 Poe?

版权:如无特殊注明,文章转载自网络,侵权请联系cnmhg168#163.com删除!文件均为网友上传,仅供研究和学习使用,务必24小时内删除。
相关推荐