Function autograd::ops::log_softmax

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

Computes log(softmax(x)) along specified axis.