ott.geometry.geometry.Geometry.apply_kernel#

Geometry.apply_kernel(scaling, eps=None, axis=0)[source]#

Apply kernel_matrix on positive scaling vector.

Parameters

  • scaling (Array) – jnp.ndarray [num_a or num_b] , scaling of size num_rows or num_cols of kernel_matrix

  • eps (Optional[float]) – passed for consistency, not used yet.

  • axis (int) – standard kernel product if axis is 1, transpose if 0.

Return type

Array

Returns

a jnp.ndarray corresponding to output above, depending on axis.