huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Laurent Mazare
2024-02-18 10:35:01 +01:00
committed by GitHub
parent 41416d2376
commit 678d44a7f6
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
candle-nn/src/conv.rs
View File

@ -109,6 +109,14 @@ impl ConvTranspose1d {
pub fn config(&self) -> &ConvTranspose1dConfig { pub fn config(&self) -> &ConvTranspose1dConfig {
&self.config &self.config
} }
pub fn weight(&self) -> &Tensor {
&self.weight
}
pub fn bias(&self) -> Option<&Tensor> {
self.bias.as_ref()
}
} }
impl crate::Module for ConvTranspose1d { impl crate::Module for ConvTranspose1d {
@ -258,6 +266,14 @@ impl ConvTranspose2d {
pub fn config(&self) -> &ConvTranspose2dConfig { pub fn config(&self) -> &ConvTranspose2dConfig {
&self.config &self.config
} }
pub fn weight(&self) -> &Tensor {
&self.weight
}
pub fn bias(&self) -> Option<&Tensor> {
self.bias.as_ref()
}
} }
impl crate::Module for ConvTranspose2d { impl crate::Module for ConvTranspose2d {