eval_shape#
- class brainpy.math.eval_shape(fun, *args, static_argnums=(), static_argnames=(), with_stack=False, **kwargs)[source]#
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.