Expose the weights and biases in transposed convolutions. (#1727)

huggingface · Feb 18, 2024 · 678d44a · 678d44a
1 parent 41416d2
commit 678d44a
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/candle-nn/src/conv.rs b/candle-nn/src/conv.rs
@@ -109,6 +109,14 @@ impl ConvTranspose1d {
     pub fn config(&self) -> &ConvTranspose1dConfig {
         &self.config
     }
+
+    pub fn weight(&self) -> &Tensor {
+        &self.weight
+    }
+
+    pub fn bias(&self) -> Option<&Tensor> {
+        self.bias.as_ref()
+    }
 }
 
 impl crate::Module for ConvTranspose1d {
@@ -258,6 +266,14 @@ impl ConvTranspose2d {
     pub fn config(&self) -> &ConvTranspose2dConfig {
         &self.config
     }
+
+    pub fn weight(&self) -> &Tensor {
+        &self.weight
+    }
+
+    pub fn bias(&self) -> Option<&Tensor> {
+        self.bias.as_ref()
+    }
 }
 
 impl crate::Module for ConvTranspose2d {