I think that’s a bug, but I can’t find documentation for torch.distributions.utils.log_sum_exp so I can’t confirm – do you know where the docs for that are?
On master there is torch.logsumexp https://github.com/pytorch/pytorch/pull/7254, but it hasn’t been documented yet. I’ll make a note that torch.distributions.utils.log_sum_exp appears to be strange.