Function log_softmax

pub fn log_softmax(
    x: impl AsRef<Array>,
    axis: impl Into<Option<i32>>,
) -> Result<Array>

Applies the Log Softmax function.

This is:

x - log_sum_exp(x, axis, true)