jax.debug.visualize_array_sharding#

jax.debug.visualize_array_sharding(arr, **kwargs)[源代码]#

可视化数组的分片。