Thank you for your quick answer!
So should I add some conv layers in __init __?
class net(nn.Module):
    def __init__(self):
        super(net, self).__init__()
        self.features = feature_layer()

    def forward(self, x):
        x = self.features(x)
        return x

I’m a beginner so the question might be kind of stupid,pardon:)