mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 18:48:51 +00:00
Add weight, bias, and hidden_size methods (#816)
* Add weight, bias methods to Conv(1|2) * Add hidden_size method to Embedding * Expose hidden_size
This commit is contained in:
@ -39,6 +39,14 @@ impl Conv1d {
|
|||||||
pub fn config(&self) -> &Conv1dConfig {
|
pub fn config(&self) -> &Conv1dConfig {
|
||||||
&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 Conv1d {
|
impl crate::Module for Conv1d {
|
||||||
@ -99,6 +107,14 @@ impl Conv2d {
|
|||||||
pub fn config(&self) -> &Conv2dConfig {
|
pub fn config(&self) -> &Conv2dConfig {
|
||||||
&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 Conv2d {
|
impl crate::Module for Conv2d {
|
||||||
|
@ -18,6 +18,11 @@ impl Embedding {
|
|||||||
pub fn embeddings(&self) -> &Tensor {
|
pub fn embeddings(&self) -> &Tensor {
|
||||||
&self.embeddings
|
&self.embeddings
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Get the hidden size of the embedding matrix
|
||||||
|
pub fn hidden_size(&self) -> usize {
|
||||||
|
self.hidden_size
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
impl crate::Module for Embedding {
|
impl crate::Module for Embedding {
|
||||||
|
Reference in New Issue
Block a user