Compute utilities for incorporating generative networks

[szmazurek]

Is your feature request related to a problem? Please describe.

Generative networks have quite different scheme for forward passes, inference and training in general. Adding new, extended versions of compute utilities is therefore necessary

Describe the solution you'd like

Adding separate modified functions that allow for complete GAN training, validation and inference

Describe alternatives you've considered

Incorporating the version modification into currently existing compute functions - altought I believe it will make the code too complex and they should be kept separate.

Additional context

Add any other context or screenshots about the feature request here.