Website Blog TensorPlay documentation

Automatic differentiation package - tensorplay.autograd Automatic Mixed Precision package - tensorplay.amp Distributed communication package - tensorplay.distributed Multiprocessing package - tensorplay.multiprocessing Quantization tensorplay tensorplay.audio tensorplay.compiler tensorplay.cuda tensorplay.fft tensorplay.futures tensorplay.hub tensorplay.library tensorplay.linalg tensorplay.nn tensorplay.nn.functional tensorplay.nn.init tensorplay.optim tensorplay.random tensorplay.sparse tensorplay.special tensorplay.stax tensorplay.utils.checkpoint tensorplay.utils.data tensorplay.vision Upstream anchor labels

tensorplay.special.entr tensorplay.special.erfcx tensorplay.special.expit tensorplay.special.gammaln tensorplay.special.log_ndtr tensorplay.special.log_softmax tensorplay.special.logsumexp tensorplay.special.modified_bessel_i0 tensorplay.special.multigammaln tensorplay.special.ndtr tensorplay.special.ndtri tensorplay.special.polygamma tensorplay.special.psi tensorplay.special.round tensorplay.special.softmax tensorplay.special.xlog1py tensorplay.special.xlogy

API symbolsspecial

Copy

View Markdown Download .md

tensorplay.special.softmax

tensorplay.special.softmax(input, dim=None, *, dtype=None)[source]

Help improve this page

Found an error, an unclear step, or a missing example?

Was this page helpful?

Suggest an edit Report an issue

tensorplay.special.round

Rounds to nearest even integer (native Tensor.round ).

tensorplay.special.xlog1py

x*log1p(y) with the convention x*log1p(-1) = 0 when x == 0.