jax.Array.mean#

abstractArray.mean(axis=None,dtype=None,out=None,keepdims=False,*,where=None)[source]#

Return the mean of array elements along a given axis.

Refer tojax.numpy.mean() for the full documentation.

Parameters:

• self (Array)

• axis (reductions.Axis)

• dtype (DTypeLike |None)

• out (None)

• keepdims (bool)

• where (ArrayLike |None)

Return type:

Array