brainstate.nn.receive_update_input

brainstate.nn.receive_update_input(cls)

The decorator to mark the object (as the before updates) to receive the input of the update function.

That is, the bef_update will receive the input of the update function:

for fun in model.bef_updates:
  fun(*args, **kwargs)
model.update(*args, **kwargs)

Return type:

object