Re-exports§
pub use activations::sigmoid;pub use activations::softmax;pub use activations::tanh;pub use conv::conv2d;pub use embedding::embedding;pub use norm::batch_norm2d;pub use norm::layer_norm;pub use pool::max_pool2d;pub use view::ReshapeBackward;
Modules§
Structs§
- AddBackward
- Fused
Matmul Backward - Matmul
Backward - Mean
Backward - MulBackward
- Relu
Backward - SubBackward
- SumBackward
- VarBackward