simpletensor.array_backend#

class simpletensor.array_backend.Backend(module)[source]#: Bases: ModuleType

simpletensor.array_backend.get_array_module(*args) → ModuleType[source]#

simpletensor.array_backend.use_cupy()[source]#: Switches array backend to cupy, if available

simpletensor.array_backend.use_numpy()[source]#: Switches array backend to numpy