jax.debug.visualize_array_sharding# jax.debug.visualize_array_sharding(arr, **kwargs)[source]# 可视化数组的分片。