def METHOD_NAME(self):
    inputs_val = np.random.random((1, N, N)).astype(aesara.config.floatX)
    inputs = aesara.shared(inputs_val)
    rfft = fft.rfft(inputs)
    f_rfft = aesara.function([], rfft)
    res_rfft = f_rfft()
    res_rfft_comp = np.asarray(res_rfft[:, :, :, 0]) + 1j * np.asarray(
        res_rfft[:, :, :, 1]
    )
    rfft_ref = np.fft.rfftn(inputs_val, axes=(1, 2))
    utt.assert_allclose(rfft_ref, res_rfft_comp, atol=1e-4, rtol=1e-4)