autopdex.utility.dict_zeros_like

autopdex.utility.dict_zeros_like(arr, **keyargs)[source]

Wrapper around zeros_like, that works also for dicts with jnp.ndarray entries.