PyTorch 是一个基于 Python 的开源机器学习库。
PyTorch 是一个基于 Python 的开源机器学习库,它主要由 Facebook 的 AI 研究实验室开发。它的两大核心特点是:
强大的 GPU 加速张量计算。
包含自动求导系统的深度神经网络。
你可以把它想象成一个非常强大、灵活的科学计算器,专门为构建和训练人工智能模型(尤其是深度学习模型)而设计。