Imported from https://github.com/shamanez/Variational-Discriminator-Bottleneck-Tensorflow-Implementation