PyRates.pyrates.backend.torch
pyrates.backend.torch module
Torch-based backend. Contains all torch-based internaö function definitions and links to torch functions.
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)[source]
Bases: pyrates.backend.base.base_backend.BaseBackend
-
get_var
(v: pyrates.backend.computegraph.ComputeVar)[source]
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)