So how does this work exactly? I see that you call inference but the result is just stored in the energy member. I mean, the outputs do not seem to depend at all on the result of the inference, which looks a bit weird to me.