ott.geometry.low_rank.LRCGeometry.apply_kernel

LRCGeometry.apply_kernel(vec, eps=None, axis=0)

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.