Pytorch袖珍手册之九

2021-08-25  本文已影响0人  深思海数_willschang
pytorch pocket reference

第五章 基于Pytorch的深度学习网络结构自主式开发

前面章节我们主要通过pytorch提供的类,函数和各种库进行模型设计及训练,在本章节中我们将通过几个代码片断来向大家展示pytorch的灵活之处,即用户可以基于pytorch的一些基础函数构建自己的新模型,激活函数,优化器等模块。

how to create your own custom deep learning components and algorithms in PyTorch.

If we take a look at the PyTorch source code, we’ll see that layers and activations are created using a functional definition anda class implementation.

书中通过对线性连接层(torch.nn.function.linear / torch.nn.Linear)的源码进行解读,说明pytorch中各layers是怎么构建的。同时自定义了一个复数形式的线性层。

自定义激活函数就比较简单,即根据激活函数算法进行编程实现。

这章节内容比较简单,到时回看书里原文即可。

原书下载地址:
我用阿里云盘分享了「OReilly.PyTorch.Pocket.R...odels.149209000X.pdf」,你可以不限速下载🚀
复制这段内容打开「阿里云盘」App 即可获取
链接:https://www.aliyundrive.com/s/NZvnGbTYr6C

上一篇 下一篇

猜你喜欢

热点阅读