stringjax_tools.auto_vectorise.auto_vmap_default_shapes

auto_vmap_default_shapes(**sample_shapes)

Temporarily update global sample-shape defaults inside a with block.

Return type:

Iterator[dict[str, Any]]