ott.geometry.geometry.Geometry.apply_kernel

ott.geometry.geometry.Geometry.apply_kernel#

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

Apply kernel_matrix on positive scaling vector.

Parameters:
  • vec (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.