首页 > 其他分享 >每天五分钟深度学习框架pytorch:nn.Module和nn.function的区别

每天五分钟深度学习框架pytorch:nn.Module和nn.function的区别

时间:2024-08-28 23:24:41浏览次数:14  
标签:function nn 区别 functional Module 参数

本文重点

前面我们学习了神经网络工具箱nn.Module,本节课程我们学习一下nn.function,我们可以暂时这样认为,基本上nn.Module所能够完成的任务,nn.function基本上都可以完成,也就是它们两个是重复的,但是它们两个还是有很大的区别,这里我们简单的进行一下介绍。

nn.Module 和 nn.functional的区别

functional可以理解为一个操作,它无法保存参数,因此它适用于一些不需要存储参数和状态信息的层,比如激活函数,droup,pooling等没有训练参数,当然这些操作也可以通过nn.Module来完成。

Module是保存了参数,因此适合于定义一层,如线性层,卷积层,也适用于定义一个网络。Module不仅包括了Function,还包括了对应的参数,以及其他函数与变量,这是Function所不具备的。

大部分nn中的类都有nn.function对应,也就是说nn都会再去调用functional,也就是说nn将functional给封装了,其区别是:nn.Module实现的laye

标签:function,nn,区别,functional,Module,参数
From: https://blog.csdn.net/huanfeng_AI/article/details/141652317

相关文章