Torch7 团队开源了 PyTorch。据官网介绍,PyTorch 是一个 Python 优先的深度学习框架,能够在强大的 GPU 加速基础上实现张量和动态神经网络。
PyTorch 是一个 Python 软件包,其提供了两种高层面的功能:
使用强大的 GPU 加速的 Tensor 计算(类似 numpy)
构建于基于 tape 的 autograd 系统的深度神经网络
如有需要,你也可以复用你最喜欢的 Python 软件包(如 numpy、scipy 和 Cython)来扩展 PyTorch。目前这个版本是早期的 Beta 版,我们很快就会加入更多的功能。
PyTorch 介绍
在粒度层面(granular level)上,PyTorch 库包含了以下组件:
使用 PyTorch 的原因通常有二:
作为 numpy 的替代,以便使用强大的 GPU;
将其作为一个能提供最大的灵活性和速度的深度学习研究平台。
进一步阐述如下:
一个支持 GPU 的 Tensor 库
如果你使用 numpy,那么你就使用过 Tensor(即 ndarray)。
PyTorch 提供了支持 CPU 和 GPU 的 Tensor,可以极大地加速计算。
我们提供了各种各样的用于加速的张量例程(tensor routine