pdft.bases.qft_code

Contents

pdft.bases.qft_code#

pdft.bases.qft_code(m, n, *, inverse=False)[source]#

Return (einsum_fn, initial_tensors) for 2D QFT on (2^m, 2^n) images.

Parameters:
Return type:

tuple[Callable[[…], Array], list[Array]]