huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 19:58:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expose the conv2d-transpose layers. (#761)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-07 07:04:52 +02:00
committed by GitHub
parent a17a7c42c1
commit 000fa00e31
2 changed files with 45 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@ -16,7 +16,10 @@ pub mod var_map;
pub use activation::Activation;
pub use batch_norm::{batch_norm, BatchNorm, BatchNormConfig};
pub use conv::{conv1d, conv2d, conv2d_no_bias, Conv1d, Conv1dConfig, Conv2d, Conv2dConfig};
pub use conv::{
conv1d, conv2d, conv2d_no_bias, conv_transpose2d, conv_transpose2d_no_bias, Conv1d,
Conv1dConfig, Conv2d, Conv2dConfig, ConvTranspose2d, ConvTranspose2dConfig,
};
pub use embedding::{embedding, Embedding};
pub use func::{func, Func};
pub use group_norm::{group_norm, GroupNorm};