PyTorch 1.2.0版正式发布,和 1.1 版本相比,新版本主要新增/改动的功能包括:
- 完善TorchScript环境,提供了新的将模型编译为TorchScrip的API
- 扩展ONNX模型的导出的支持
- 增强对Transformer架构的模块级支持
- 此外,TensorBoard现在也能快速开始使用了
PyTorch 是一个 Torch7 团队开源的 Python 优先的深度学习框架,提供两个高级功能:
-
强大的 GPU 加速 Tensor 计算(类似 numpy)
-
构建基于 tape 的自动升级系统上的深度神经网络
你可以重用你喜欢的 python 包,如 numpy、scipy 和 Cython ,在需要时扩展 PyTorch。