brainpy.math.training_environment

class brainpy.math.training_environment(dt=None, x64=None, complex_=None, float_=None, int_=None, bool_=None, batch_size=1)

Environment with the training mode.

This is a short-cut context setting for an environment with the training mode. It is equivalent to:

>>> import brainpy.math as bm
>>> with bm.environment(mode=bm.training_mode):
>>>   pass

__init__(dt=None, x64=None, complex_=None, float_=None, int_=None, bool_=None, batch_size=1)

Methods

__init__([dt, x64, complex_, float_, int_, ...])
clone()