本文重点
前面我们学习了神经网络工具箱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