add_module(name,module)

将一个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)