TensorPlay
API symbolsspecial
Copy
View MarkdownDownload .md

tensorplay.special.logsumexp

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

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

Ask DeepWiki