log_mean

pmrf.math.aggregations.log_mean(x: Array) → Array

Computes the mean of the log of all elements of the input array.

This is equivalent to the log of the geometric mean, but is often more numerically stable.

Parameters:

x (jnp.ndarray) – Input array.

Returns:

The geometric mean of the input array.

Return type:

jnp.ndarray