jaxsnn.discrete.modules.linear.Projection

class jaxsnn.discrete.modules.linear.Projection(generator: Callable[[int, int], jaxsnn.discrete.types.ProjectionFunctions], parameters: Dict[str, Any])

Bases: jaxsnn.discrete.types.BaseModule[jaxsnn.discrete.types.ProjectionFunctions]

A projection module for the discrete paradigm.

__init__(generator: Callable[[int, int], jaxsnn.discrete.types.ProjectionFunctions], parameters: Dict[str, Any]) → None

Initialize self. See help(type(self)) for accurate signature.

Methods

__init__(generator, parameters)

Initialize self.

Attributes

Functions

alias of jaxsnn.discrete.types.ProjectionFunctions

generator: Callable[[int, int], jaxsnn.discrete.types.ProjectionFunctions]