2.4 PyTorch简介

在编写本书时,PyTorch是第三大最受欢迎的深度学习程序框架。尽管与TensorFlow相比,PyTorch还相对比较新,但它的受欢迎程度一直在上升。PyTorch的一个有趣之处在于,它允许一些TensorFlow不允许的定制。此外,PyTorch还支持FacebookTM

尽管本书涵盖了TensorFlow和Keras,但我认为重要的是所有人都要记住PyTorch是一个很好的替代方案,而且它看起来非常类似于Keras。仅作为参考,下面是前述浅层神经网络在PyTorch中编码时的样子:

二者的相似之处有很多。此外,从Keras到PyTorch的转换对有学习兴趣的读者来说应该不是太难,而且将来可能会成为一个很好的技能。然而,目前社区的兴趣主要集中在TensorFlow及其衍生物上,尤其是Keras。如果你想要了解更多关于PyTorch的入门知识和基本原理,可能会发现这本书比较有用(Paszke,A.,et.al.,2017)。