fix for #6700

                weight /= weight.mean()

            elif use_weight:

                #If an image does not have a alpha channel, add a ones weight map anyway so we can stack it later

                weight = torch.ones([channels] + latent_size)

                weight = torch.ones(latent_sample.shape)

            else:

                weight = None