torch.nn与torch.nn.functional的区别
torch.nn:
类似于nn.Conv2d,nn.MaxPool2d等功能均是以层的方式构建,主要在网络层搭建的时候使用
必须搭载nn.Module容器
torch.nn.functional:
类似于nn.functional.conv2d,nn.functional.max_pool2d等功能均是以函数方式构建
对于输入获取输出时使用
选择哪种方式,取决于自己的代码习惯
...
原创
2021-10-14 20:08:35 ·
218 阅读 ·
0 评论