diffusion_rs_common::nn::ops

Function dropout

source
pub fn dropout(xs: &Tensor, drop_p: f32) -> Result<Tensor>