Module params

Source
Expand description

this module provides the ParamsBase type for the library, which is used to define the parameters of a neural network. Parameters for constructing neural network models. This module implements parameters using the ParamsBase struct and its associated types. The ParamsBase struct provides:

  • An (n) dimensional weight tensor as ArrayBase
  • An (n-1) dimensional bias tensor as ArrayBase

The associated types follow suite with the ndarray crate, each of which defines a different style of representation for the parameters.

Modules§

error
iter
params

Structs§

ParamsBase
The ParamsBase struct is a generic container for a set of weights and biases for a model. The implementation is designed around the ArrayBase type from the ndarray crate, which allows for flexible and efficient storage of multi-dimensional arrays.

Enums§

ParamsError

Type Aliases§

ArcParams
a type alias for shared parameters
CowParams
a type alias for borrowed parameters
Params
a type alias for owned parameters
ParamsView
a type alias for an immutable view of the parameters
ParamsViewMut
a type alias for a mutable view of the parameters
RawMutParams
a mutable raw view of the parameters; internally uses a mutable pointer
RawViewParams
a raw view of the parameters; internally uses a constant pointer