PyTorch是一个Python包,它提供了两个主要的高级功能:类似于NumPy的张量计算,但具有强大的GPU加速,以及建立在基于磁带的autograd系统上的深度神经网络。该包可以无缝地整合用户首选的Python包,如NumPy,SciPy和Cython,以根据需要扩展PyTorch。

PyTorch 是一个包含各种组件的库,包括:

  • torch,一个类似于 NumPy 的张量库,但具有强大的 GPU 支持。该库还包括torch.autograd,这是一个基于磁带的自动微分库,支持Torch中的所有可微张量操作。
  • torch.jit,一个编译堆栈(TorchScript),从PyTorch代码生成可序列化和可优化的模型。
  • Torch.nn,一个与Autograd深度集成的神经网络库,旨在提供最大的灵活性。
  • torch.multiprocessing,它提供Python多处理功能,但跨进程共享Torch张量的神奇内存,使其对数据加载和Hogwild训练特别有用。
  • torch.utils,包括DataLoader和其他实用程序功能,以方便用户。

PyTorch通常以以下两种方式之一使用。

  • 作为NumPy的替代品,以利用GPU的强大功能
  • 作为一个深度学习研究平台,提供最大的灵活性和速度。