Coupled KL Divergence does not support batches

Photrek / Nonlinear-Statistical-Coupling

Apache License 2.0

5 stars 1 forks source link

Coupled KL Divergence does not support batches #23

Closed Kevin-Chen0 closed 3 years ago

Kevin-Chen0 commented 3 years ago

I have created two MultivariateCoupledNormal distributions, both of bach_size=32 batches and dim=2. However, I get the following error when performing coupled_kl_divergence:

jkclem commented 3 years ago

This issue is resolved and can be closed.