diffusion_rs_common::nn::ops

Function softmax_last_dim

Source 
pub fn softmax_last_dim(xs: &Tensor) -> Result<Tensor>