728x90
3차원이나 4차원이 들어와야하는데 2차원이 들어와서 생긴 문제
z = torch.randn((real_img.shape[0], config.latent_size), device=config.device, dtype=torch.float32)
위의 코드를 아래의 코드로 고쳐서 해결
z = torch.randn((real_img.shape[0], config.latent_size,1,1), device=config.device, dtype=torch.float32)
728x90