How to use the spectral.T2FFT.T2FFT.analyze function in spectral

To help you get started, we’ve selected a few spectral examples, based on popular ways it is used in public projects.

Secure your code as it's written. Use Snyk Code to scan source code in minutes - no build needed - and fix issues immediately.

github AMLab-Amsterdam / lie_learn / lie_learn / spectral / SE2FFT.py View on Github external
fs = []
    fhs = []

    for i in range(36):
        t = i * 2 * np.pi / 36.
        R = np.array([[np.cos(-t), -np.sin(-t)],
                      [np.sin(-t), np.cos(-t)]])
        RC = np.einsum('ij,abj->abi', R, C - np.array([p0, q0])[None, None, :]) + np.array([p0, q0])[None, None, :]
        #Rfr = map_wrap(f1.real, RC.transpose(2, 0, 1))
        #Rfi = map_wrap(f1.imag, RC.transpose(2, 0, 1))
        #Rf = Rfr + Rfi * 1j

        Rf = np.exp(1j * 2 * np.pi * (5 * RC[..., 0]) / nx)
        Rfh = shift_fft(Rf)
        Rfh = T2FFT.analyze(Rf)

        fs.append(Rf)
        fhs.append(Rfh)

    return fs, fhs