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

jax.Array.size#

property Array.size_: int_[source]#

The total number of elements in the array.