jax.Array.mean

内容

jax.Array.mean#

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

返回沿给定轴的数组元素的平均值。

有关完整文档,请参阅 jax.numpy.mean()

参数:
  • self (Array)

  • axis (reductions.Axis)

  • dtype (DTypeLike | None)

  • out (None)

  • keepdims (bool)

  • where (ArrayLike | None)

返回类型:

Array