pytorch中的nn.Module和nn.functional模块

一、nn.Module类

通过继承 nn.Module 转换成为类的实现形式,并直接封装在 nn 模块下,其中包含了常见的卷积层、池化层、以及损失函数等,下面将会介绍该模块下的一些函数,通常我们会import torch.nn as nn,以下都基于nn来表述:
1.激活函数:
nn.Relu
nn.Sigmoid
nn.Tanh
nn.Softmax
2.模型层:
nn.Linear
nn.Conv2d
nn.MaxPool2d
nn.Dropout2d
nn.Embedding
3.损失函数:
nn.BCELoss
nn.MSELoss
nn.CrossEntropyLoss

二、nn.functional模块

该模块的功能和nn模块差不多,包括损失函数、网络层之类的,通过该模块去调用相应的函数其实也是转到了nn模块下,所以该方法不常用,我也不常用,所以不做具体介绍。

之后会通过实战实例来详细介绍这写模块的函数调用方法,毕竟手撕代码比理论学习来得更加清晰,更加迅速。

你可能感兴趣的:(pytorch)