Bases: discrete_wavelet_expansion
Source code in tinybig/expansion/wavelet_expansion.py
| class dog_wavelet_expansion(discrete_wavelet_expansion):
def __init__(self, name: str = 'dog_wavelet_expansion', a: float = 1.0, b: float = 1.0, sigma_1: float = 1.0, sigma_2: float = 2.0, *args, **kwargs):
super().__init__(name=name, *args, **kwargs)
self.wavelet = dog_wavelet(a=a, b=b, sigma_1=sigma_1, sigma_2=sigma_2)
|