Function dfdx::tensor_ops::softmax

source · [−]
pub fn softmax<T: Reduce1<-1>>(t: T) -> T
Expand description

Computes the softmax function. Equivalent to exp(log_softmax(t)).

Pytorch equivalent: t.softmax(-1)

Related functions: logsumexp(), log_softmax()