PyTorch官网介绍
PyTorch是一个开源的Python机器学习库,基于Torch库,底层由C++实现,是目前人工智能领域的事实标准。它最初由Facebook的人工智能研究团队开发,现在属于Linux基金会的一部分。经过多年的发展和斗争,PyTorch已经远胜于TensorFlow,前者生态繁盛,而后者在逐渐凋零。
PyTorch特点介绍
PyTorch主要有两大特征:
(1)类似于NumPy的张量计算,能在 GPU 或 MPS 等硬件加速器上加速;
(2)基于带自动微分系统的深度神经网络。
另外,PyTorch技术栈十分丰富,包括torch.autograd、torch.nn、torch.optim等子模块,也包含多种损失函数,例如 MSE(均方误差或者称为"L2范数")、交叉熵损失和负熵似然损失(对分类器有用)等。