From 758366160e26a493fb5a1d151dcdab9c8abf99c8 Mon Sep 17 00:00:00 2001 From: Kirpal Grewal <45569241+KGrewal1@users.noreply.github.com> Date: Fri, 8 Mar 2024 07:18:01 +0000 Subject: add clone to candle dropout (#1814) --- candle-nn/src/ops.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'candle-nn') 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, } -- cgit v1.2.3