torch.nn:
类似于nn.Conv2d
,nn.MaxPool2d
等功能均是以层的方式构建,主要在网络层搭建的时候使用
必须搭载nn.Module
容器
torch.nn.functional:
类似于nn.functional.conv2d
,nn.functional.max_pool2d
等功能均是以函数方式构建
对于输入获取输出时使用
选择哪种方式,取决于自己的代码习惯
torch.nn:
类似于nn.Conv2d
,nn.MaxPool2d
等功能均是以层的方式构建,主要在网络层搭建的时候使用
必须搭载nn.Module
容器
torch.nn.functional:
类似于nn.functional.conv2d
,nn.functional.max_pool2d
等功能均是以函数方式构建
对于输入获取输出时使用
选择哪种方式,取决于自己的代码习惯