stringjax_tools.auto_vectorise.set_auto_vmap_default_shapes

set_auto_vmap_default_shapes(**sample_shapes)

Update global default exact sample-shape specifications.

These defaults are used only when auto_vmap() is called with validate_shapes=True.

Return type:

dict[str, Any]