z_dim is initialized as 64 in simple GAN code but shows some matrix multiplication error in a first linear layer of the Generator class.