Hello!
That’s strange…
So, you are calling __init__ only once and then apply forward on the same input multiple times without doing any optimization or changing the weights in any way, right?
Edit: I don’t know if this has something to do with your issue but the class you declared has something wrong: the methods of the class should have self as first argument. Moreover usually you want to subclass nn.Module and calling super in the constructor.