Skip to main content

softmax

Function softmax 

Source
pub fn softmax(input: &Tensor, dim: isize) -> Result<Tensor>
Expand description

Softmax along specified dimension