BackendManager.change_backend#

BackendManager.change_backend(backend_name, **backend_kwargs)[source]#

Change the current computation backend.

Parameters:
backend_namestr

Name of the new backend that should be used.

**backend_kwargsdict, optional

Parameters passed to __init__ method of backend.

Raises:
NotImplementedError

If no backend is found with the provided name.

Examples

>>> from tme.backends import backend as be
>>> be.change_backend("cupy")