EligibilityTrace

class braintrace.EligibilityTrace(value, name=None, **metadata)

The state for storing the eligibility trace during the computation of online learning algorithms.

Examples

An eligibility trace wraps an array-valued state, just like any other brainstate.ShortTermState:

>>> import jax.numpy as jnp
>>> import braintrace
>>>
>>> trace = braintrace.EligibilityTrace(jnp.zeros((3, 4)))
>>> trace.value.shape
(3, 4)