pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

AI 简介

PyTorch 是一个开源的深度学习框架，提供支持 GPU 加速的张量计算和基于动态计算图的自动微分功能。其核心包括类 NumPy 的 torch 张量库、tape-based autograd 自动求导系统、torch.nn 神经网络模块以及 TorchScript 模型编译与部署能力。采用 Python 优先设计，支持命令式编程、灵活的模型构建与调试，并可无缝集成 NumPy、SciPy 等生态工具。适用于学术研究、工业级模型训练与推理、教学实验及边缘设备（如 Jetson）上的 AI 开发场景。