I wouldn’t recommend to recreate new modules in the forward, as they would be reinitialized in each forward pass and thus not trained.
I’m not sure what the actual use case is and where these shapes would be used. Are you trying to pass them to the transposed conv layers?