jaxsnn.discrete.modules.input

Classes

DenseData	alias of jax.Array
Parameter	alias of jax.Array
SourcePopulation(generator, …)	A source population module for the discrete paradigm.

Functions

jaxsnn.discrete.modules.input.Input(size: int) → jaxsnn.discrete.types.SourcePopulation

Create an input layer descriptor.

It returns None for the generator because it’s is not needed.

Parameters

size – Number of neurons in the input layer.

Returns

A SourcePopulation instance representing the input layer.