Python kl_divergence Examples

Example #1

 def loss(self):
     return measures.kl_divergence(self.x, self.reconstruction)

Example #2

File: qmf.py Project: ziyouzizai111/google-research

 def loss(self, x, rows=None, cols=None):
     reconstruction = self.sliced_reconstruction(rows, cols)
     reg = self.regularization(rows=rows, cols=cols)
     return measures.kl_divergence(x, reconstruction) + reg