Skip to content

shannon_wavelet

Bases: discrete_wavelet

Source code in tinybig/koala/signal_processing/wavelet.py 
64
65
66
67
68
69
class shannon_wavelet(discrete_wavelet):
    def __init__(self, name: str = 'shannon_wavelet', *args, **kwargs):
        super().__init__(name=name, *args, **kwargs)

    def psi(self, tau: torch.Tensor):
        return (torch.sin(2*torch.pi*tau) - torch.sin(torch.pi*tau))/(torch.pi*tau)