class_docstr: Array base class for JAX
  
  ``jax.Array`` is the public interface for instance checks and type annotation
  of JAX arrays and tracers. Its main applications are in instance checks and
  type annotations; for example::
  
    x = jnp.arange(5)
    isinstance(x, jax.Array)  # returns True both inside and outside traced functions.
  
    def f(x: Array) -> Array:  # type annotations are valid for traced and non-traced types.
      return x
  
  ``jax.Array`` should not be used directly for creation of arrays; instead you
  should use array creation routines offered in :mod:`jax.numpy`, such as
  :func:`jax.numpy.array`, :func:`jax.numpy.zeros`, :func:`jax.numpy.ones`,
  :func:`jax.numpy.full`, :func:`jax.numpy.arange`, etc.
class_docstr_tok: ["Array","base","class","for","JAX","``","jax.Array","``","is","the","public","interface","for","instance","checks","and","type","annotation","of","JAX","arrays","and","tracers",".","Its","main","applications","are","in","instance","checks","and","type","annotations",";","for","example",":",":","x","=","jnp.arange","(","5",")","isinstance","(","x",",","jax.Array",")","#","returns","True","both","inside","and","outside","traced","functions",".","def","f","(","x",":","Array",")","-",">","Array",":","#","type","annotations","are","valid","for","traced","and","non-traced","types",".","return","x","``","jax.Array","``","should","not","be","used","directly","for","creation","of","arrays",";","instead","you","should","use","array","creation","routines","offered","in",":","mod",":","`","jax.numpy","`",",","such","as",":","func",":","`","jax.numpy.array","`",",",":","func",":","`","jax.numpy.zeros","`",",",":","func",":","`","jax.numpy.ones","`",",",":","func",":","`","jax.numpy.full","`",",",":","func",":","`","jax.numpy.arange","`",",","etc","."]
docstr: The total number of elements in the array.
docstr_tok: ["The","total","number","of","elements","in","the","array","."]
code: def size(self) -> int:
      """The total number of elements in the array."""
  
file_path: jax/jax/_src/basearray.py
imports: from __future__ import annotations
  import abc
  import numpy
  from typing import Any, Union
  from collections.abc import Sequence
label_desc: Use image node_id 3 for calling the Array obj's underlying member method code with example usage: obj.size() without return types