similarity.py 文件源码

def sym_heat_similarity_matrix(X, sigma):
    """
    Defines the self similarity matrix using the heat kernel
    :param X:
    :param sigma:
    :return:
    """
    D = sym_distance_matrix(X, X, self_similarity=True)
    return torch.exp(-D ** 2 / (sigma ** 2))