Function kernel

ot/utils.py:42–49

r"""Compute kernel matrix

(x1, x2, method="gaussian", sigma=1, **kwargs)

def kernel(x1, x2, method="gaussian", sigma=1, **kwargs):
    r"""Compute kernel matrix"""

    nx = get_backend(x1, x2)

    if method.lower() in ["gaussian", "gauss", "rbf"]:
        K = nx.exp(-dist(x1, x2) / (2 * sigma**2))
    return K


def laplacian(x):