Need minus sign in reverse log density computation
jaanli opened this issue · 0 comments
jaanli commented
Line 114 in 34a34a0
This currently matches
Line 130 in 34a34a0
The forward computation is on
Line 113 in 34a34a0
u = (inputs - m) * torch.exp(-a)
The inverse of this is
inputs = u * torch.exp(a) + m
So the log determinant jacobian of the inverse is a.sum(-1, keepdim=True)
not -a.sum(-1, keepdim=True)
.
Might be missing something here!