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

jax.Array.nbytes#

abstract property Array.nbytes_: int_[source]#

Total bytes consumed by the elements of the array.