jax.Array.sharding — JAX documentation (original) (raw)

jax.Array.sharding#

property Array.sharding_: Sharding_[source]#

The sharding for the array.