ott.geometry.geometry.Geometry.apply_kernel#
- Geometry.apply_kernel(vec, eps=None, axis=0)[source]#
Apply
kernel_matrix
on positive scaling vector.- Parameters:
- Return type:
- Returns:
a jnp.ndarray corresponding to output above, depending on axis.
Apply kernel_matrix
on positive scaling vector.
a jnp.ndarray corresponding to output above, depending on axis.