Function candle_nn::ops::log_softmax

pub fn log_softmax<D: Dim>(xs: &Tensor, d: D) -> Result<Tensor>