self.pool = nn.ModuleList()
时间: 2023-05-14 13:05:34 浏览: 95
对Pytorch中nn.ModuleList 和 nn.Sequential详解
这是一个 Python 代码片段,用于创建一个空的 nn.ModuleList 对象并将其赋值给 self.pool 变量。nn.ModuleList 是 PyTorch 中的一个类,用于存储 nn.Module 对象的列表。通常用于管理神经网络中的子模块。
阅读全文
对Pytorch中nn.ModuleList 和 nn.Sequential详解