pixel-constrained-cnn-pytorch
pixel-constrained-cnn-pytorch copied to clipboard
Fixed log_likelihood()
Fixed a bug in log_likelihood() caused by not selecting indices correctly. Also vectorised it for efficiency.