huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 11:08:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expose more conv1d functions/structs. (#1726)

Browse Source
This commit is contained in:
Laurent Mazare
2024-02-17 18:50:55 +01:00
committed by GitHub
parent 5ebcfeaf0f
commit 41416d2376
2 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -302,6 +302,22 @@ pub fn conv1d(
Ok(Conv1d::new(ws, Some(bs), cfg)) Ok(Conv1d::new(ws, Some(bs), cfg))
} }
pub fn conv1d_no_bias(
in_channels: usize,
out_channels: usize,
kernel_size: usize,
cfg: Conv1dConfig,
vb: crate::VarBuilder,
) -> Result<Conv1d> {
let init_ws = crate::init::DEFAULT_KAIMING_NORMAL;
let ws = vb.get_with_hints(
(out_channels, in_channels / cfg.groups, kernel_size),
"weight",
init_ws,
)?;
Ok(Conv1d::new(ws, None, cfg))
}
pub fn conv_transpose1d( pub fn conv_transpose1d(
in_channels: usize, in_channels: usize,
out_channels: usize, out_channels: usize,

5
candle-nn/src/lib.rs
View File

@ -19,8 +19,9 @@ pub mod var_map;
pub use activation::{prelu, Activation, PReLU}; pub use activation::{prelu, Activation, PReLU};
pub use batch_norm::{batch_norm, BatchNorm, BatchNormConfig}; pub use batch_norm::{batch_norm, BatchNorm, BatchNormConfig};
pub use conv::{ pub use conv::{
conv1d, conv2d, conv2d_no_bias, conv_transpose2d, conv_transpose2d_no_bias, Conv1d, conv1d, conv1d_no_bias, conv2d, conv2d_no_bias, conv_transpose1d, conv_transpose1d_no_bias,
Conv1dConfig, Conv2d, Conv2dConfig, ConvTranspose2d, ConvTranspose2dConfig, conv_transpose2d, conv_transpose2d_no_bias, Conv1d, Conv1dConfig, Conv2d, Conv2dConfig,
ConvTranspose1d, ConvTranspose1dConfig, ConvTranspose2d, ConvTranspose2dConfig,
}; };
pub use embedding::{embedding, Embedding}; pub use embedding::{embedding, Embedding};
pub use func::{func, func_t, Func, FuncT}; pub use func::{func, func_t, Func, FuncT};