eval_shape

class brainpy.math.eval_shape(fun, *args, static_argnums=(), static_argnames=(), with_stack=False, **kwargs)

Compute the shape/dtype of fun without any FLOPs.

Parameters:

• fun (Callable) – The callable function.

• *args – The positional arguments.

• **kwargs – The keyword arguments.

• with_stack (bool) – Whether evaluate the function within a local variable stack.

• static_argnums (Sequence[int]) – The static argument indices.

• static_argnames (Sequence[str]) – The static argument names.

Returns:

The variable stack and the functional returns.