diff options
| -rw-r--r-- | candle-nn/src/layer_norm.rs | 8 |
1 files changed, 8 insertions, 0 deletions
diff --git a/candle-nn/src/layer_norm.rs b/candle-nn/src/layer_norm.rs index 7617fc6c..26800a7b 100644 --- a/candle-nn/src/layer_norm.rs +++ b/candle-nn/src/layer_norm.rs @@ -95,6 +95,14 @@ impl LayerNorm { eps, } } + + pub fn weight(&self) -> &Tensor { + &self.weight + } + + pub fn bias(&self) -> Option<&Tensor> { + self.bias.as_ref() + } } impl crate::Module for LayerNorm { |