Module dfdx::nn

High level neural network building blocks such as Linear, activations, and tuples as feedforward networks.

Saving and loading model parameters is done using SaveToNpz and LoadFromNpz. All modules provided here implement it, including tuples. So you can call SaveToNpz::save() to save the module to a .npz zip file format, and then LoadFromNpz::load() to load the weights.

Structs

Abs

Unit struct that impls Module as calling abs() on input.

Cos

Unit struct that impls Module as calling cos() on input.

Dropout

A Module that calls dropout() in Module::forward() with probability self.p.

Exp

Unit struct that impls Module as calling exp() on input.

Linear

A linear transformation of the form xW + b, where W is a matrix, x is a vector or matrix, and b is a vector. If x is a matrix this does matrix multiplication.

Ln

Unit struct that impls Module as calling ln() on input.

ReLU

Unit struct that impls Module as calling relu() on input.

Sigmoid

Unit struct that impls Module as calling sigmoid() on input.

Sin

Unit struct that impls Module as calling sin() on input.

Sqrt

Unit struct that impls Module as calling sqrt() on input.

Square

Unit struct that impls Module as calling square() on input.

Tanh

Unit struct that impls Module as calling tanh() on input.

Enums

NpzError

Error that can happen while loading data from a .npz zip archive.

Traits

LoadFromNpz

Something that can be loaded from a .npz file (which is a zip file).

Module

A unit of a neural network. Acts on the generic Input and produces Module::Output.

SaveToNpz

Something that can be saved to a .npz (which is a .zip).