diffusion_rs_common::nn::ops

Function inplace_softmax_last_dim

source 
pub fn inplace_softmax_last_dim(xs: &mut Tensor) -> Result<()>