PyTorch是一个Python包,它提供了Tensor计算(像NumPy一样),具有强大的GPU加速和基于磁带的自动记录系统构建的深度神经网络,这个项目可以实现快速、灵活的实验和高效的生产。
软件特色
PyTorch由torch(Tensor库)、torch.autograd(基于磁带的自动分化库)、torch.jit(编译栈[TorchScript])、torch.nn(神经网络库)、torch.multiprocessing(Python多处理)和torch.utils(DataLoader和其他实用函数)组成。PyTorch可以作为Numpy的替代品,也可以作为深度学习研究平台,提供最佳的灵活性和速度。
TorchScript
分布式培训
工具和图书馆
云计算合作伙伴
原生ONNX支持
C++前端
机器学习框架