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

• API Reference
• jax.Array.sharding

jax.Array.sharding#

property Array.sharding_: Sharding_[source]#

The sharding for the array.