burn_tensor/tensor/ops/modules/
mod.rs

1/// Module with convolution operations.
2pub mod conv;
3
4/// Module with cat operation
5pub(crate) mod cat;
6/// Module with repeat operation
7pub(crate) mod repeat_dim;
8/// Module with unfold operations.
9pub(crate) mod unfold;
10
11/// Module with pooling operations.
12pub mod pool;
13
14mod base;
15
16pub use base::*;