jax.nn.logmeanexp#
- jax.nn.logmeanexp(x, axis=None, where=None, keepdims=False)[源代码]#
对数平均指数。
计算函数
\[\text{logmeanexp}(x) = \log \frac{1}{n} \sum_{i=1}^n \exp x_i = \text{logsumexp}(x) - \log n\]- 参数:
x (ArrayLike) – 输入数组。
axis (Axis) – 沿其进行约简的轴或轴。
where (ArrayLike | None) – 包含在约简中的元素。可选。
keepdims (bool) – 保留输入的维度。
- 返回:
一个数组。
- 返回类型:
另请参阅