tensorplay.special._elementwise.logsumexp

tensorplay.special._elementwise.logsumexp(input, dim=None, keepdim=False, *, out=None)[source]

Log-sum-exp along dim, computed with the max-shift trick.