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)