Sorry I am a bit confused here. Let’s say l10 belongs to this UNet structure, before it was a linear module plus one learnable parameter, by “previous layer”, you mean I input my data,  go through all the layers before l10, and directly used the output of l10 to compute loss and back propagation?