Source code for pennylane.transforms.combine_global_phases

# Copyright 2018-2024 Xanadu Quantum Technologies Inc.

# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at

#     http://www.apache.org/licenses/LICENSE-2.0

# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

"""
Provides a transform to combine all ``qml.GlobalPhase`` gates in a circuit into a single one applied at the end.
"""

import pennylane as qml
from pennylane.tape import QuantumScript, QuantumScriptBatch
from pennylane.transforms import transform
from pennylane.typing import PostprocessingFn


@transform
def combine_global_phases(tape: QuantumScript) -> tuple[QuantumScriptBatch, PostprocessingFn]:
    """Combine all ``qml.GlobalPhase`` gates into a single ``qml.GlobalPhase`` operation.

    This transform returns a new circuit where all ``qml.GlobalPhase`` gates in the original circuit (if exists)
    are removed, and a new ``qml.GlobalPhase`` is added at the end of the list of operations with its phase
    being a total global phase computed as the algebraic sum of all global phases in the original circuit.

    Args:
        tape (QNode or QuantumScript or Callable): the input circuit to be transformed.

    Returns:
        qnode (QNode) or quantum function (Callable) or tuple[List[QuantumScript], function]:
        the transformed circuit as described in :func:`qml.transform <pennylane.transform>`.

    **Example**

    Suppose we want to combine all the global phase gates in a given quantum circuit.
    The ``combine_global_phases`` transform can be used to do this as follows:

    .. code-block:: python3

        dev = qml.device("default.qubit", wires=3)

        @qml.transforms.combine_global_phases
        @qml.qnode(dev)
        def circuit():
            qml.GlobalPhase(0.3, wires=0)
            qml.PauliY(wires=0)
            qml.Hadamard(wires=1)
            qml.CNOT(wires=(1,2))
            qml.GlobalPhase(0.46, wires=2)
            return qml.expval(qml.X(0) @ qml.Z(1))

    To check the result, let's print out the circuit:

    >>> print(qml.draw(circuit)())
    0: ──Y─────GlobalPhase(0.76)─┤ ╭<X@Z>
    1: ──H─╭●──GlobalPhase(0.76)─┤ ╰<X@Z>
    2: ────╰X──GlobalPhase(0.76)─┤
    """

    has_global_phase = False
    phi = 0
    operations = []
    for op in tape.operations:
        if isinstance(op, qml.GlobalPhase):
            has_global_phase = True
            phi += op.parameters[0]
        else:
            operations.append(op)

    if has_global_phase:
        with qml.QueuingManager.stop_recording():
            operations.append(qml.GlobalPhase(phi=phi))

    new_tape = tape.copy(operations=operations)

    def null_postprocessing(results):
        """A postprocesing function returned by a transform that only converts the batch of results
        into a result for a single ``QuantumScript``.
        """
        return results[0]

    return (new_tape,), null_postprocessing

_modules/pennylane/transforms/combine_global_phases

 

Download Python script

 

Download Notebook

 

View on GitHub