在一个网络1中途加入另一个网络2

网络1：

class net1(nn.Module):
    def __init__(self):
        super(net1, self).__init__()
        self.f1 = nn.Linear(6,6)
        self.f2 = nn.Linear(6,6)
        self.f3 = nn.Linear(6,6)
        self.net2=net2()
    def forward(self,x):
        x=self.f1(x)
        x=self.f2(x)
        x=self.net2(x)
        x=self.f3(x)
        return  x

网络2：

class net2(nn.Module):
    def __init__(self):
        super(net2, self).__init__()
        self.f1 = nn.Linear(6, 6)
        self.f2 = nn.Linear(6, 6)
        self.f3 = nn.Linear(6, 6)

    def forward(self, x):
        x = self.f1(x)
        x = self.f2(x)
        x = self.f3(x)
        return x

测试：

if __name__ == '__main__':

    tudui=net1()
    input=torch.ones(64,6)
    output=tudui(input)
    print(output.shape)