Do you mean from torch.logsumexp?
‘https://pytorch.org/docs/stable/generated/torch.logsumexp.html 6’
If so, we do have torch::logsumexp(tensor, {}, bool keepdim = false) where {} is a list of dimension you want to reduce.
’