torch.nn.Module.modules()
torch.nn.Module.modules()
狼啸风云
修改于 2022-09-02 22:26:31
修改于 2022-09-02 22:26:31
Returns an iterator over all modules in the network.(返回网络中所有模块的迭代器。)
Yields
Module – a module in the network
>>> l = nn.Linear(2, 2)
>>> net = nn.Sequential(l, l)
>>> for idx, m in enumerate(net.modules()):
print(idx, '->', m)
0 -> Sequential(
(0): Linear(in_features=2, out_features=2, bias=True)
(1): Linear(in_features=2, out_features=2, bias=True)
)
1 -> Linear(in_features=2, out_features=2, bias=True)Submodules assigned in this way will be registered, and will have their parameters converted too when you call to(), etc
本文参与 腾讯云自媒体分享计划，分享自作者个人站点/博客。
原始发表：2020-05-07 ，如有侵权请联系 cloudcommunity@tencent.com 删除
评论
登录后参与评论
推荐阅读