23. PyTorch
Last updated
Last updated
neural-network
autograd
numpy
deep-learning
tensor
PyTorch 是一个 Python 优先的深度学习框架,也是使用 GPU 和 CPU 优化的深度学习张量库,能够在强大的 GPU 加速基础上实现张量和动态神经网络。其前身是 Torch,主要语言接口为 Lua。PyTorch 和 TensorFlow、MXNet、Caffe2 一样,是非常底层的框架;也正如 TensorFlow 是谷歌官方框架,MXNet 是亚马逊官方框架,背后支持 PyTorch 的则是 Facebook。
同时,PyTorch 还是一个 Python 软件包,其提供了两种高层面的功能:
使用强大的 GPU 加速的 Tensor 计算(类似 numpy)
构建于基于 tape 的 autograd 系统的深度神经网络
通常,人们使用 PyTorch 的原因通常有二:
作为 numpy 的替代,以便使用强大的 GPU
将其作为一个能提供最大的灵活性和速度的深度学习研究平台