Thanks for your input @JuanFMontesinos . I’m all for that sort of convenience, I just don’t think it should fit under the “forward” function of a “neural_network”.Module. If we’re going to go down the path of doing pre/post processing on the GPU then there should be separate types of modules that handle it. It’s not just a cosmetic thing. It also means that when I want to do something custom with the pre/post processing I need to dive into the source code and isolate the neural network from the rest manually.