PyRates.pyrates.backend.base

pyrates.backend.base module

default PyRates backend. Contains all pyrates-internal function definitions and links to numpy functions.

pyrates.backend.base.base_backend module

Contains wrapper classes for different backends that are needed by the parser module.

A new backend needs to implement the following methods: - __init__ - run - add_var - add_op - add_layer

Currently supported backends: - Numpy: BaseBackend. - Tensorflow: TensorflowBackend. - Fortran: FortranBackend (experimental).

class pyrates.backend.base.base_backend.BaseBackend(ops: Optional[Dict[str, str]] = None, imports: Optional[List[str]] = None, **kwargs)

Bases: pyrates.backend.base.base_backend.CodeGen

Default backend class. Transforms all network equations into their numpy equivalents. Based on a Python code generator.

add_var_hist(lhs: str, delay: Union[pyrates.backend.computegraph.ComputeVar, float], state_idx: str, **kwargs)

add_var_update(lhs: pyrates.backend.computegraph.ComputeVar, rhs: str, lhs_idx: Optional[str] = None, rhs_shape: Optional[tuple] = ())

clear()

Deletes all code lines from the memory of the generator.

create_index_str(idx: Union[str, int, tuple], separator: str = ',', apply: bool = True, **kwargs) → Tuple[str, dict]

static expr_to_str(expr: str, args: tuple)

static finalize_idx_str(var: pyrates.backend.computegraph.ComputeVar, idx: str)

generate_func(func_name: str, to_file: bool = True, **kwargs)

generate_func_head(func_name: str, state_var: str = 'y', return_var: str = 'dy', func_args: list = None, add_hist_func: bool = False)

generate_func_tail(rhs_var: str = 'dy')

get_fname(f: str) → tuple

static get_hist_func(y: numpy.ndarray)

get_op(name: str, **kwargs) → dict

get_var(v: pyrates.backend.computegraph.ComputeVar)

static register_vars(variables: list)

run(func: Callable, func_args: tuple, T: float, dt: float, dts: float, outputs: dict, solver: str, **kwargs) → dict

static to_file(fn: str, **kwargs)

class pyrates.backend.base.base_backend.CodeGen

Bases: object

Generates python code. Can add code lines, line-breaks, indents and remove indents.

add_code_line(code_str)

Add code line string to code.

add_indent()

Add an indent to the code.

add_linebreak()

Add a line-break to the code.

clear()

Deletes all code lines from the memory of the generator.

generate()

Generates a single code string from its history of code additions.

remove_indent()

Remove an indent to the code.

pyrates.backend.base.base_funcs module

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

pyrates.backend.base.base_funcs.identity(x)

pyrates.backend.base.base_funcs.index_1d(x, idx)

pyrates.backend.base.base_funcs.index_2d(x, idx1, idx2)

pyrates.backend.base.base_funcs.index_axis(x, idx=None, axis=0)

pyrates.backend.base.base_funcs.index_range(x, idx1, idx2)

pyrates.backend.base.base_funcs.past(x, tau)