Function burn::tensor::activation::log_softmax

pub fn log_softmax<const D: usize, B>(
    tensor: Tensor<B, D>,
    dim: usize,
) -> Tensor<B, D>
where
    B: Backend,

Applies the log softmax function on the input tensor along the given dimension.

log_softmax(x_i) = log(softmax(x_i)) = log(exp(x_i) / sum_j(exp(x_j)))

Notes

The dimension argument dim specifies the dimension along which the function will be computed. It must in the range of 0 and D-1.