jaxsnn.event.types.LayeredEventT

class jaxsnn.event.types.LayeredEventT(*args, **kwargs)

Bases: Protocol

__init__(*args, **kwargs)

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

Methods

__init__(*args, **kwargs)

Initialize self.

idx: jax.Array

internal: jax.Array

layer_idx: jax.Array

time: jax.Array