PyTorch官网

撰写于:  浏览:1022 次

PyTorch官网介绍

https://pytorch.org/

PyTorch是一个开源的Python机器学习库,基于Torch库,底层由C++实现,是目前人工智能领域的事实标准。它最初由Facebook的人工智能研究团队开发,现在属于Linux基金会的一部分。经过多年的发展和斗争,PyTorch已经远胜于TensorFlow,前者生态繁盛,而后者在逐渐凋零。

PyTorch特点介绍

PyTorch主要有两大特征:
(1)类似于NumPy的张量计算,能在 GPU 或 MPS 等硬件加速器上加速;
(2)基于带自动微分系统的深度神经网络。

另外,PyTorch技术栈十分丰富,包括torch.autograd、torch.nn、torch.optim等子模块,也包含多种损失函数,例如 MSE(均方误差或者称为"L2范数")、交叉熵损失和负熵似然损失(对分类器有用)等。

关注公众号,每天学一个AI知识

    友情链接