2.1　PyTorch概述

2.1.1　什么是PyTorch

什么是PyTorch？其实，PyTorch可以拆分成两部分：Py和Torch。Py就是Python，Torch是一个有大量机器学习算法支持的科学计算框架。Lua语言简洁高效，但由于其过于小众，用的人不是很多。考虑到Python在人工智能领域的领先地位，以及其生态的完整性和接口的易用性，几乎任何框架都不可避免地要提供Python接口。终于，2017年，Torch的幕后团队使用Python重写了Torch的很多内容，推出了PyTorch，并提供了Python接口。此后，PyTorch成为最流行的深度学习框架之一。

直白地说，PyTorch可以看成一个Python库，可以像NumPy、Pandas一样被Python所调用。PyTorch与NumPy的功能是类似的，可以把PyTorch看成应用在神经网络里的NumPy，而且是加入了GPU支持的NumPy。

2.1.2　为什么使用PyTorch

深度学习框架那么多，为什么我们这么青睐PyTorch呢？我们先来看看有哪些机构在使用PyTorch吧，如图2-1所示。

图2-1　使用PyTorch的机构

由图2-1可知，PyTorch已经被Twitter、CMU和Salesforce等多个机构使用。这足以说明PyTorch是好用的，而且是值得推广的。

PyTorch实际上是NumPy的替代工具，而且支持GPU，带有高级功能，可以用来搭建和训练深度神经网络。只要熟悉NumPy、Python以及基础的深度学习概念，PyTorch就非常容易上手。这就是PyTorch备受青睐的重要原因之一。