Function autograd::ops::log_softmax [] [src]

pub fn log_softmax(x: &Tensor, axis: isize) -> Tensor

Log softmax function.

Computes softmax(x) along specified axis and takes logarithm of it. axis can be negative.