brainstate.transform.eval_shape

Contents

brainstate.transform.eval_shape#

brainstate.transform.eval_shape(f, *args, **kwargs)[source]#

Evaluate the shape of the output of a function.

Return type:

TypeVar(A)