Module dfdx::nn

Neural network building blocks such as Linear and impls for tuples as feedforward networks.

Structs

Abs

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

Cos

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

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.

Traits

Module

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