The shape of the `latent` argument is not, as users might guess, `(1, 16, h//8, w//8)`, but `(1, 128, h//16, w//16)`. This should be documented.