Add weight and bias functions to LayerNorm (#1306)

huggingface · Nov 9, 2023 · e669747 · e669747
1 parent 73d02f4
commit e669747
Showing 1 changed file with 8 additions and 0 deletions.
diff --git 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 {