Expand description
Neural network building blocks such as Linear and impls for tuples as feedforward networks.
Structs
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.
Traits
A unit of a neural network. Acts on the generic Input
and produces Module::Output
.