jax.debug.visualize_array_sharding

jax.debug.visualize_array_sharding#

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

可视化数组的分片。