将一个child module 添加到当前modle, 被添加的module可以通过name属性来获取
import torch.nn as nn
class Model(nn.Module):
def __init__(self):
super(Model, self).__init__()
self.add_module("conv", nn.Conv2d(10, 20, 4))
#self.conv = nn.Conv2d(10, 20, 4) is the same as above
model = Model()
print(model.conv)#Conv2d(10, 20, kernel_size=(4, 4), stride=(1, 1)