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

jax.Array.shape#

property Array.shape_: tuple[int, ...]_[source]#

The shape of the array.