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: Dict[str, str] | None = None, imports: List[str] | None = None, **kwargs)[source]¶
Bases:
BaseBackend
- get_var(v: ComputeVar)[source]¶
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)¶