一个用于构建和训练神经网络的深度学习框架。
PyTorch 像实验室白板,公式刚写一半就能上手改;哪层不对,研究员擦了就重来。
它适合快速搭模型、做实验和训练网络,很多论文原型都先在这儿落地。
Framework
PyTorch 是深度学习框架的一种,主打灵活开发与训练。
Automatic Differentiation
它内置自动求导,能自动计算训练所需梯度。
GPU
PyTorch 常把计算放到 GPU 上,加快模型训练与实验。
Cuda
它经常配合 CUDA 使用,把模型计算交给显卡执行。