diffusion_rs_common::nn::ops

Function log_softmax

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