parameters.Constant

parameters.Constant(obj: Any) → RP_Constant

Identify an initialisation argument as a constant (non-trainable) parameter

Examples

>>> mod = LIFJax(1)
>>> mod.parameters('taus')
{'tau_mem': DeviceArray([0.02], dtype=float32),
 'tau_syn': DeviceArray([[0.02]], dtype=float32)}
>>> mod.simulation_parameters('taus')
{}

>>> mod = LIFJax(1, tau_mem = Constant(10e-3))
>>> mod.parameters('taus')
{'tau_syn': DeviceArray([[0.02]], dtype=float32)}
>>> mod.simulation_parameters('taus')
{'tau_mem': DeviceArray(0.01, dtype=float32)}

Parameters:

obj (Any) – The initialisation object to wrap

Returns:

A wrapped object, of the same class as obj.