本文介绍了PyTorch中nn模块和functional模块的使用场景。nn模块主要用于构建网络层，其组件如nn.Conv2d和nn.MaxPool2d以层的形式存在，需要在nn.Module容器中使用。而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等功能均是以函数方式构建
对于输入获取输出时使用
选择哪种方式，取决于自己的代码习惯
扫一扫
评论
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
查看更多评论
添加红包
