Get some layers in a pytorch model that is not defined by nn.Sequential

Question

I have a network defined below.

class model_dnn_2(nn.Module):
    def __init__(self):
        super(model_dnn_2, self).__init__()
        self.flatten = Flatten()