Expand description
Multi-layer perceptron (MLP) core.
The low-level API is intentionally allocation-free:
Mlp::forwardwrites activations into a reusableScratchand returns a slice.Mlp::backwardwrites gradients into a reusableGradients.
Shape mismatches are treated as programmer error and will panic via assert!.
Structsยง
- Batch
Backprop Scratch - Reusable buffers for
Mlp::backward_batch. - Batch
Scratch - Reusable buffers for
Mlp::forward_batch. - Gradients
- Parameter gradients for an
Mlp(overwrite semantics). - Mlp
- A feed-forward multi-layer perceptron composed of dense layers.
- Scratch
- Reusable buffers for
Mlp::forward. - Trainer
- Reusable buffers for training a specific
Mlp.