xnn.ffnn.models.geometry.geometric_mean

xnn.ffnn.models.geometry.geometric_mean(x, y)

Geometric mean with a differentiable zero (for zero-LJ hydrogens).

Parameters:

• x (Tensor) – Non-negative parameter values, gathered per pair.

• y (Tensor) – Non-negative parameter values, gathered per pair.

Returns:

sqrt(x * y), exactly zero (with zero gradient) where the product vanishes.

Return type:

Tensor