ott.geometry.geometry.Geometry.apply_kernel
ott.geometry.geometry.Geometry.apply_kernel#
- Geometry.apply_kernel(scaling, eps=None, axis=0)[source]#
Apply
kernel_matrix
on positive scaling vector.This function applies the ground geometry’s kernel, to perform either output = K v (1) output = K’u (2) where K is [num_a, num_b]
- Parameters
- Return type
- Returns
a jnp.ndarray corresponding to output above, depending on axis.