
【Torch巩固复习记录】torch.nn
torch.nn中的nn全称为neural network,意思是神经网络,是torch中构建神经网络的模块。在看过一些资料后,我对nn这个模块是这么理解的:这个模块的基础是nn.Module,是实现的层(layer)是一个特殊的类。都是由class Layer(nn.Module)定义,会自动提取可学习的参数。ps:functinal模块中的函数就是一个函数,没有可学习参数。因此,nn就是pytorch用来提供各类预定义好的模型层的库,有什么需要的模型层在这个库里面找就好了。











