PyTorch包

2024 年 8 月 29 日 | 阅读 2 分钟

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

序号名称描述
1.Torchtorch 包包含多维张量的数据结构,并定义了对这些张量的数学运算。
2.torch.Tensor此包是一个多维矩阵,其中包含单一数据类型的元素。
3.张量属性
a) torch.dtype它是一个对象,表示torch.Tensor 的数据类型。
b) torch.device它是一个对象,表示将分配 torch.Tensor 的设备。
c) torch.layout它是一个对象,表示 toch.Tensor 的内存布局。
4.类型信息可以通过 torch.iinfo 或 torch.finfo 访问 torch.dtype 的数值属性。
1) torch.finfo它是一个对象,表示浮点 torch.dtype 的数值属性。
2) torch.iinfo它是一个对象,表示整数 torch.dtype 的数值属性。
5.torch.sparseTorch 支持 COO(坐标)格式的稀疏张量,它可以有效地存储和处理大多数元素为零的张量。
6.torch.cudaTorch 支持 CUDA 张量类型,它实现与 CPU 张量相同的功能,但对于计算,它们使用 GPU。
7.torch.Storagetorch.Storage 是单一数据类型的连续一维数组。
8.torch.nn此包为我们提供了更多类和模块来实现和训练神经网络。
9.torch.nn.functional此包具有与 torch.nn 类似的函数类。
10.torch.optim此包用于实现各种优化算法。
11.torch.autogard此包提供类和函数来实现任意标量值函数的自动微分。
12.torch.distributed此包支持三个后端,每个后端都具有不同的功能。
13.torch.distribution此包允许我们构建随机计算图,以及用于优化的随机梯度估计器
14.torch.hub它是一个预训练模型存储库,旨在促进研究的可重复性。
15.torch.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.onnxONNX 导出器是一个基于跟踪的导出器,这意味着它通过执行一次模型并导出在此运行期间实际运行的运算符来运行

引用

https://pytorch.ac.cn/docs/stable/index.html