diff --git a/candle-nn/src/ops.rs b/candle-nn/src/ops.rs index fdd67142..88d1b3d6 100644 --- a/candle-nn/src/ops.rs +++ b/candle-nn/src/ops.rs @@ -74,7 +74,7 @@ pub fn dropout(xs: &Tensor, drop_p: f32) -> Result { xs * mask } -#[derive(Debug)] +#[derive(Clone, Debug)] pub struct Dropout { drop_p: f32, }