jax.Array.mean

abstract Array.mean(axis=None, dtype=None, out=None, keepdims=False, *, where=None)

沿著指定軸返回陣列元素的平均值。

完整文件請參閱 jax.numpy.mean()。

參數:

• self (Array)

• axis (reductions.Axis)

• dtype (DTypeLike | None)

• out (None)

• keepdims (bool)

• where (ArrayLike | None)

返回類型:

Array