Skip to content

anisotropic_rbf_kernel_interdependence

Bases: numerical_kernel_based_interdependence

Source code in tinybig/interdependence/numerical_kernel_interdependence.py 
139
140
141
142
class anisotropic_rbf_kernel_interdependence(numerical_kernel_based_interdependence):
    def __init__(self, name: str = 'anisotropic_rbf_kernel_interdependence', a_vector: Union[torch.Tensor, np.array] = None, a_scalar: float = 1.0, *args, **kwargs):
        anisotropic_rbf_kernel_func = functools.partial(anisotropic_rbf_kernel, a_vector=a_vector, a_scalar=a_scalar)
        super().__init__(kernel=anisotropic_rbf_kernel_func, name=name, *args, **kwargs)