#2 · 主分类： 深度学习框架
pytorch
autograd
deep-learning
gpu
machine-learning
neural-network
numpy
python
tensor
Python中的张量和动态神经网络，具有强大的GPU加速
项目最后更新：08/29/26
GitHub Stars
102.7K
Forks数量
29.0K
贡献者数量
6.9K
许可证
Other
收录理由
PyTorch 是许多 Python 团队在构建和训练神经网络时的首选库。它的张量操作用起来像 NumPy，却能跑在 GPU 上；基于 tape 的自动求导系统允许你在训练过程中随时调整模型结构，不必像静态图那样重新构建。这种灵活性让它成为研究工作和快速迭代新架构的常备工具。torch.nn 模块提供了现成的层和损失函数，TorchScript 以及周边生态则能帮你把训练好的模型推向生产环境。无论你是刚起步做深度学习项目，还是正在维护现有代码，大概率都会直接跟这套 API 打交道，早点熟悉它的构建选项会省下不少功夫。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容，欢迎 联系我们推荐选题或投稿。