diff options
Diffstat (limited to 'candle-nn/src/batch_norm.rs')
| -rw-r--r-- | candle-nn/src/batch_norm.rs | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/candle-nn/src/batch_norm.rs b/candle-nn/src/batch_norm.rs index 05904859..8cfc6740 100644 --- a/candle-nn/src/batch_norm.rs +++ b/candle-nn/src/batch_norm.rs @@ -109,6 +109,10 @@ impl BatchNorm { &self.running_var } + pub fn eps(&self) -> f64 { + self.eps + } + pub fn weight_and_bias(&self) -> Option<(&Tensor, &Tensor)> { self.weight_and_bias.as_ref().map(|v| (&v.0, &v.1)) } |