jax.debug.visualize_array_sharding#

jax.debug.visualize_array_sharding(arr, **kwargs)[source]#

視覺化陣列的分片。