geometric_mean

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

Computes the geometric mean over all elements of the input array.

Parameters:

x (jnp.ndarray) – Input array.

Returns:

The geometric mean of the input array.

Return type:

jnp.ndarray