Formatting#

Utility functions for formatting and displaying array shapes and dtypes.

nabla.utils.formatting.format_dtype(dtype)[source]#

Format dtype for display.

nabla.utils.formatting.format_shape_and_dtype(array)[source]#

Format shape and dtype in JAX style with batch_dims numbers in blue and everything else in purple.