I want to create a network which has two types of trainable sublayers, with Type1 fixed and shared between Type2.
class Net(nn.Module):      
    def __init__(self):
        super(Net, self).__init__()
        self.Layer1 = nn.Type1
        self.Layer2 = nn.Type2_1
        self.Layer3 = nn.Type1
        self.Layer4 = nn.Type2_2
        .......

How can create this network and define a forward pass for depth K, i.e., the model will have one Type1 layer and K Type2 layers