safe_to_summarize

safe_to_summarize#

penzai.treescope.ndarray_summarization.safe_to_summarize(array: np.ndarray | jax.Array | Any) bool[source]#

Checks if the array is safe to summarize (not a tracer and not replicated).