Expand description
Machine learning kernels for GPU
Re-exports§
pub use activation::GeluKernel;pub use activation::LeakyReluKernel;pub use activation::ReluKernel;pub use activation::SigmoidKernel;pub use activation::SwishKernel;pub use activation::TanhKernel;
Modules§
- activation
- Activation function kernels for neural networks
- pooling
- Pooling operation kernels for neural networks
- softmax
- Softmax activation function kernel