eval_shape

Contents

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.