Typing

FunctionalSample module-attribute

FunctionalSample = Callable[[Float[Array, 'N D']], Float[Array, 'N B']]

Type alias for functions representing $B$ samples from a model, to be evaluated on any set of $N$ inputs (of dimension $D$) and returning the evaluations of each (potentially approximate) sample draw across these inputs.