博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
Function与Module的差异与应用场景
阅读量:5123 次
发布时间:2019-06-13

本文共 541 字,大约阅读时间需要 1 分钟。

Function与Module都可以对pytorch进行自定义拓展,使其满足网络的需求,但这两者还是有十分重要的不同:

Function一般只定义一个操作,因为其无法保存参数,因此适用于激活函数、pooling等操作;Module是保存了参数,因此适合于定义一层,如线性层,卷积层,也适用于定义一个网络

Function需要定义三个方法:__init__, forward, backward(需要自己写求导公式);Module:只需定义__init__和forward,而backward的计算由自动求导机制构成
可以不严谨的认为,Module是由一系列Function组成,因此其在forward的过程中,Function和Variable组成了计算图,在backward时,只需调用Function的backward就得到结果,因此Module不需要再定义backward。
Module不仅包括了Function,还包括了对应的参数,以及其他函数与变量,这是Function所不具备的

文:http://www.ylsjwang.com/dianshiju/62.html

转载于:https://www.cnblogs.com/sjc9009/p/9382466.html

你可能感兴趣的文章
Python6 模块
查看>>
P3377 【模板】左偏树(可并堆)
查看>>
Djang 用户登录
查看>>
Java同步锁——lock与synchronized 的区别【转】
查看>>
洛谷-校门外的树-数组
查看>>
Python--网络编程-----文件传输简单版本
查看>>
3 使用模块
查看>>
解决前端页面运行出现乱码的现象
查看>>
CF 208E. Blood Cousins [dsu on tree 倍增]
查看>>
趣谈面试(一)
查看>>
Quart2D setNeedsDisplay
查看>>
设计模式之策略设计模式
查看>>
SQL2005 安装时 “性能监视器计数器要求(错误)” 解决方案
查看>>
杂-lowbit
查看>>
724#寻找数组的中心索引
查看>>
用Python做一个飞机大战游戏
查看>>
登录注册
查看>>
HDU 5429 Geometric Progression
查看>>
mysql 视图入门
查看>>
[原创]Eclipse Memory Analyzer tool(MAT)工个使用介绍
查看>>