Module iter_params

Module iter_params 

Source
Expand description

Iterators for parameters within a neural network

Structsยง

ParamsIter
The ParamsIter defines a iterator over owned parameters within a neural network by zipping together an axis iterator over the columns of the weights and an iterator over the bias.
ParamsIterMut
The ParamsIterMut type provides a mutable iterator over the parameters of a neural network layer by zipping together a mutable axis iterator over the columns of the weights and a mutable iterator over the bias.