How would would take latent code with dimensions (batch_size,K) and then do some element wise multiplication to turn that matrix into dim (batch_size,channels, height, width) where height and width will be N by N matrix essentially and you fill K dimensions into the height, width dimensions. Since K will be less than N by N matrix or (h,w) there will be n-k zero entries.