flatten_for_tensor_contraction

View source on GitHub

Flatten a (composite) bloq as much as possible to enable efficient tensor contraction.

qualtran.simulation.tensor.flatten_for_tensor_contraction(
    bloq: qualtran.Bloq,
    full_flatten: bool = True
) -> qualtran.CompositeBloq

Args

bloq

The bloq to flatten.

full_flatten

Whether to completely flatten the bloq into the smallest possible bloqs. Otherwise, stop flattening if custom tensors are encountered.