jax.Array
=========

.. currentmodule:: jax

.. autoclass:: Array

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~Array.__init__
      ~Array.addressable_data
      ~Array.all
      ~Array.any
      ~Array.argmax
      ~Array.argmin
      ~Array.argpartition
      ~Array.argsort
      ~Array.astype
      ~Array.choose
      ~Array.clip
      ~Array.compress
      ~Array.conj
      ~Array.conjugate
      ~Array.copy
      ~Array.copy_to_host_async
      ~Array.cumprod
      ~Array.cumsum
      ~Array.diagonal
      ~Array.dot
      ~Array.flatten
      ~Array.item
      ~Array.max
      ~Array.mean
      ~Array.min
      ~Array.nonzero
      ~Array.prod
      ~Array.ptp
      ~Array.ravel
      ~Array.repeat
      ~Array.reshape
      ~Array.round
      ~Array.searchsorted
      ~Array.sort
      ~Array.squeeze
      ~Array.std
      ~Array.sum
      ~Array.swapaxes
      ~Array.take
      ~Array.to_device
      ~Array.trace
      ~Array.transpose
      ~Array.var
      ~Array.view
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~Array.T
      ~Array.addressable_shards
      ~Array.at
      ~Array.committed
      ~Array.device
      ~Array.dtype
      ~Array.flat
      ~Array.global_shards
      ~Array.imag
      ~Array.is_fully_addressable
      ~Array.is_fully_replicated
      ~Array.itemsize
      ~Array.mT
      ~Array.nbytes
      ~Array.ndim
      ~Array.real
      ~Array.shape
      ~Array.sharding
      ~Array.size