diff options
| author | Kirpal Grewal <45569241+KGrewal1@users.noreply.github.com> | 2024-03-08 07:18:01 +0000 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-03-08 08:18:01 +0100 |
| commit | 758366160e26a493fb5a1d151dcdab9c8abf99c8 (patch) | |
| tree | e06f2c44bfb5b22a17a3bce0b51ef6732a66a38b /candle-nn | |
| parent | 0a3487a776a08171f39ec7784c0a66ff94f98ec5 (diff) | |
| download | candle-758366160e26a493fb5a1d151dcdab9c8abf99c8.tar.gz candle-758366160e26a493fb5a1d151dcdab9c8abf99c8.tar.bz2 candle-758366160e26a493fb5a1d151dcdab9c8abf99c8.zip | |
add clone to candle dropout (#1814)
Diffstat (limited to 'candle-nn')
| -rw-r--r-- | candle-nn/src/ops.rs | 2 |
1 files changed, 1 insertions, 1 deletions
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<Tensor> { xs * mask } -#[derive(Debug)] +#[derive(Clone, Debug)] pub struct Dropout { drop_p: f32, } |