On master there is torch.logsumexp https://github.com/pytorch/pytorch/pull/7254 51, but it hasn’t been documented yet. I’ll make a note that torch.distributions.utils.log_sum_exp appears to be strange.