ott.math.utils.softmin

Contents

ott.math.utils.softmin#

ott.math.utils.softmin(x, gamma, axis=None) = <jax._src.custom_derivatives.custom_vjp object>[source]#

Soft-min operator.

Parameters:
  • x (Array) – Input data.

  • gamma (float) – Smoothing parameter \(> 0\).

  • axis (Optional[int]) – Axis or axes over which to operate. If None, use flattened input.

Return type:

Array

Returns:

The soft minimum.