1.1.1.5. PyRates.pyrates.backend.torch

1.1.1.5.1. pyrates.backend.torch module

Torch-based backend. Contains all torch-based internaö function definitions and links to torch functions.

1.1.1.5.2. pyrates.backend.torch.torch_backend module

Wraps torch such that it’s low-level functions can be used by PyRates to create and simulate a compute graph.

class pyrates.backend.torch.torch_backend.TorchBackend(ops: Optional[Dict[str, str]] = None, imports: Optional[List[str]] = None, **kwargs)

Bases: BaseBackend

get_var(v: ComputeVar)

1.1.1.5.3. pyrates.backend.torch.torch_funcs module

Contains torch function definitions that may be used for PyRates model equations.

pyrates.backend.torch.torch_funcs.sigmoid(x)