As @Nikronic mentioned the kl_div requires a to be  log-probabilities  and the target  b  to be  probability. correct me if I’m wrong, but I think sampling from the two distributions is not going to give  log-probabilities and probabilities