eazygrad.nn.Module¶

class eazygrad.nn.Module[source]¶

Bases: object

forward(*args: Any) → Any[source]¶: Apply the forward pass of the module to the input x

parameters() → list[Any][source]¶

register_params(params: Any) → None[source]¶