Compile#

compile_function(inputs, outputs[, cache, ...])

Compile a sympy function to a pytensor function.

compile_to_pytensor_function(inputs, ...)

Convert a sympy function to a pytensor function using pytensor.function().

compile_for_scipy(outputs[, mode])

Compile a pytensor graph into a callable that accepts all inputs as keyword arguments.

pack_and_compile(outputs, ss_nodes[, ...])

Replace SS variable scalars with x_flat[i] indexing and compile.

sympy_to_pytensor(inputs, outputs[, cache, cse])

Convert sympy expressions to pytensor graph nodes.

Function Wrappers#

dictionary_return_wrapper(f, outputs)

Wrap a function that returns a numpy array to instead return a SymbolDictionary.