API reference
latest (dev)
Copy
Latest development documentation · Updated 2026-10-15
tensorplay.special._elementwise API
Functions 4
log_softmax
functionFull reference ↗- tensorplay.special._elementwise.log_softmax(input, dim=None, *, dtype=None)[source]
log(softmax(input, dim)), evaluated without the intermediate exp.
logsumexp
functionFull reference ↗- tensorplay.special._elementwise.logsumexp(input, dim=None, keepdim=False, *, out=None)[source]
Log-sum-exp along
dim, computed with the max-shift trick.
round
functionFull reference ↗- tensorplay.special._elementwise.round(input)[source]
Rounds to nearest even integer (native
Tensor.round).
softmax
functionFull reference ↗- tensorplay.special._elementwise.softmax(input, dim=None, *, dtype=None)[source]
Normalized exponential along
dim(default: the last dim).
Help improve this page
Found an error, an unclear step, or a missing example?
Was this page helpful?