diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2024-02-17 18:50:55 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-02-17 18:50:55 +0100 |
| commit | 41416d23762e6cf21aa6e1c3adb6d457f15d7071 (patch) | |
| tree | 5f0c164b3dcfb70241cb48215c88100d13d5bfff /candle-nn/src/conv.rs | |
| parent | 5ebcfeaf0f5af69bb2f74385e8d6b020d4a3b8df (diff) | |
| download | candle-41416d23762e6cf21aa6e1c3adb6d457f15d7071.tar.gz candle-41416d23762e6cf21aa6e1c3adb6d457f15d7071.tar.bz2 candle-41416d23762e6cf21aa6e1c3adb6d457f15d7071.zip | |
Expose more conv1d functions/structs. (#1726)
Diffstat (limited to 'candle-nn/src/conv.rs')
| -rw-r--r-- | candle-nn/src/conv.rs | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/candle-nn/src/conv.rs b/candle-nn/src/conv.rs index b1168405..6734ab1f 100644 --- a/candle-nn/src/conv.rs +++ b/candle-nn/src/conv.rs @@ -302,6 +302,22 @@ pub fn conv1d( 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( in_channels: usize, out_channels: usize, |