Measurements

PennyLane can extract different types of measurement results from quantum devices: the expectation of an observable, its variance, samples of a single measurement, or computational basis state probabilities.

For example, the following circuit returns the expectation value of the PauliZ observable on wire 1:

def my_quantum_function(x, y):
    qml.RZ(x, wires=0)
    qml.CNOT(wires=[0, 1])
    qml.RY(y, wires=1)
    return qml.expval(qml.PauliZ(1))

The available measurement functions are

expval(op)

Expectation value of the supplied observable.

sample([op, wires])

Sample from the supplied observable, with the number of shots determined from the dev.shots attribute of the corresponding device, returning raw samples.

counts([op, wires, all_outcomes])

Sample from the supplied observable, with the number of shots determined from the dev.shots attribute of the corresponding device, returning the number of counts for each sample.

var(op)

Variance of the supplied observable.

probs([wires, op])

Probability of each computational basis state.

state()

Quantum state in the computational basis.

density_matrix(wires)

Quantum density matrix in the computational basis.

vn_entropy(wires[, log_base])

Von Neumann entropy of the system prior to measurement.

mutual_info(wires0, wires1[, log_base])

Mutual information between the subsystems prior to measurement:

purity(wires)

The purity of the system prior to measurement.

classical_shadow(wires[, seed])

The classical shadow measurement protocol.

shadow_expval(H[, k, seed])

Compute expectation values using classical shadows in a differentiable manner.

Note

All measurement functions support analytic differentiation, with the exception of sample(), counts(), and classical_shadow(), as they return stochastic results.

Combined measurements

Quantum functions can also return combined measurements of multiple observables. If the observables are not qubit-wise-commuting, then multiple device executions may occur behind the scenes. Non-commuting oberservables can not be simultaneously measured in conjunction with non-observable type measurements such as sample(), counts(), probs(), state(), and density_matrix().

def my_quantum_function(x, y):
    qml.RZ(x, wires=0)
    qml.CNOT(wires=[0, 1])
    qml.RY(y, wires=1)
    return qml.expval(qml.PauliZ(0)), qml.var(qml.PauliX(0))

You can also use list comprehensions, and other common Python patterns:

def my_quantum_function(x, y):
    qml.RZ(x, wires=0)
    qml.CNOT(wires=[0, 1])
    qml.RY(y, wires=1)
    return [qml.expval(qml.PauliZ(i)) for i in range(2)]

As a full example of combined measurements, let us look at a Bell state \((|00\rangle + |11\rangle)/\sqrt{2}\), prepared by a Hadamard and CNOT gate.

import pennylane as qml
from pennylane import numpy as np

dev = qml.device("default.qubit", wires=2, shots=1000)

@qml.qnode(dev)
def circuit():
    qml.Hadamard(wires=0)
    qml.CNOT(wires=[0, 1])
    return qml.sample(qml.PauliZ(0)), qml.sample(qml.PauliZ(1))

The combined PauliZ-measurement of the first and second qubit returns a tuple of two arrays, each containing the measurement results of the respective qubit. sample() returns 1000 samples per observable as defined on the device.

>>> results = circuit()
>>> results[0].shape
(1000,)
>>> results[1].shape
(1000,)

Since the two qubits are maximally entangled, the measurement results always coincide, and the lists are therefore equal:

>>> np.all(result[0] == result[1])
True

Tensor observables

PennyLane supports measuring the tensor product of observables, by using the @ notation. For example, to measure the expectation value of \(Z\otimes I \otimes X\):

def my_quantum_function(x, y):
    qml.RZ(x, wires=0)
    qml.CNOT(wires=[0, 1])
    qml.RY(y, wires=1)
    qml.CNOT(wires=[0, 2])
    return qml.expval(qml.PauliZ(0) @ qml.PauliX(2))

Note that we don’t need to declare the identity observable on wire 1; this is implicitly assumed.

The tensor observable notation can be used inside all measurement functions that accept observables as arguments, including expval(), var(), and sample().

Counts

To avoid dealing with long arrays for the larger numbers of shots, one can use counts() rather than sample(). This performs the same measurement as sampling, but returns a dictionary containing the possible measurement outcomes and the number of occurrences for each, rather than a list of all outcomes.

The previous example will be modified as follows:

dev = qml.device("default.qubit", wires=2, shots=1000)

@qml.qnode(dev)
def circuit():
    qml.Hadamard(wires=0)
    qml.CNOT(wires=[0, 1])
    return qml.counts(qml.PauliZ(0)), qml.counts(qml.PauliZ(1))

After executing the circuit, we can directly see how many times each measurement outcome occurred:

>>> circuit()
({-1: 496, 1: 504}, {-1: 496, 1: 504})

Similarly, if the observable is not provided, the count of the observed computational basis state is returned.

dev = qml.device("default.qubit", wires=2, shots=1000)

@qml.qnode(dev)
def circuit():
    qml.Hadamard(wires=0)
    qml.CNOT(wires=[0, 1])
    return qml.counts()

And the result is:

>>> circuit()
{'00': 495, '11': 505}

By default, only observed outcomes are included in the dictionary. The kwarg all_outcomes=True can be used to display all possible outcomes, including those that were observed 0 times in sampling.

For example, we could run the previous circuit with all_outcomes=True:

dev = qml.device("default.qubit", wires=2, shots=1000)

@qml.qnode(dev)
def circuit():
    qml.Hadamard(wires=0)
    qml.CNOT(wires=[0, 1])
    return qml.counts(all_outcomes=True)
>>> result = circuit()
>>> print(result)
{'00': 518, '01': 0, '10': 0, '11': 482}

Note: For complicated Hamiltonians, this can add considerable overhead time (due to the cost of calculating eigenvalues to determine possible outcomes), and as the number of qubits increases, the length of the output dictionary showing possible computational basis states grows rapidly.

If counts are obtained along with a measurement function other than sample(), a tuple is returned to provide differentiability for the outputs of QNodes.

@qml.qnode(dev)
def circuit():
    qml.Hadamard(wires=0)
    qml.CNOT(wires=[0,1])
    qml.X(1)
    return qml.expval(qml.PauliZ(0)),qml.expval(qml.PauliZ(1)), qml.counts()
>>> circuit()
(-0.036, 0.036, {'01': 482, '10': 518})

Probability

You can also train QNodes on computational basis probabilities, by using the probs() measurement function. The function can accept either specified wires or an observable that rotates the computational basis.

def my_quantum_function(x, y):
    qml.RZ(x, wires=0)
    qml.CNOT(wires=[0, 1])
    qml.RY(y, wires=1)
    qml.CNOT(wires=[0, 2])
    return qml.probs(wires=[0, 1])

For example:

>>> dev = qml.device("default.qubit", wires=3)
>>> qnode = qml.QNode(my_quantum_function, dev)
>>> qnode(0.56, 0.1)
array([0.99750208, 0.00249792, 0.        , 0.        ])

The returned probability array uses lexicographical ordering, so corresponds to a \(99.75\%\) probability of measuring state \(|00\rangle\), and a \(0.25\%\) probability of measuring state \(|01\rangle\).

Mid-circuit measurements and conditional operations

PennyLane allows specifying measurements in the middle of the circuit. Quantum functions such as operations can then be conditioned on the measurement outcome of such mid-circuit measurements:

def my_quantum_function(x, y):
    qml.RY(x, wires=0)
    qml.CNOT(wires=[0, 1])
    m_0 = qml.measure(1)

    qml.cond(m_0, qml.RY)(y, wires=0)
    return qml.probs(wires=[0])

Deferred measurements

A quantum function with mid-circuit measurements (defined using measure()) and conditional operations (defined using cond()) can be executed by applying the deferred measurement principle. In the example above, we apply the RY rotation if the mid-circuit measurement on qubit 1 yielded 1 as an outcome, otherwise doing nothing for the 0 measurement outcome.

PennyLane implements the deferred measurement principle to transform conditional operations with the defer_measurements() quantum function transform. The deferred measurement principle provides a natural method to simulate the application of mid-circuit measurements and conditional operations in a differentiable and device-independent way. Performing true mid-circuit measurements and conditional operations is dependent on the quantum hardware and PennyLane device capabilities.

transformed_qfunc = qml.transforms.defer_measurements(my_quantum_function)
transformed_qnode = qml.QNode(transformed_qfunc, dev)
pars = np.array([0.643, 0.246], requires_grad=True)
>>> transformed_qnode(*pars)
tensor([0.90165331, 0.09834669], requires_grad=True)

The decorator syntax applies equally well:

@qml.qnode(dev)
@qml.defer_measurements
def qnode(x, y):
    (...)

Resetting wires

Wires can be reused as normal after making mid-circuit measurements. Moreover, a measured wire can also be reset to the \(|0 \rangle\) state by setting the reset keyword argument of measure() to True.

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

@qml.qnode(dev)
def func():
    qml.PauliX(1)
    m_0 = qml.measure(1, reset=True)
    qml.PauliX(1)
    return qml.probs(wires=[1])

Executing this QNode:

>>> func()
tensor([0., 1.], requires_grad=True)

Conditional operators

Users can create conditional operators controlled on mid-circuit measurements using cond(). We can also specify an outcome when defining a conditional operation:

@qml.qnode(dev)
@qml.defer_measurements
def qnode_conditional_op_on_zero(x, y):
    qml.RY(x, wires=0)
    qml.CNOT(wires=[0, 1])
    m_0 = qml.measure(1)

    qml.cond(m_0 == 0, qml.RY)(y, wires=0)
    return qml.probs(wires=[0])

pars = np.array([0.643, 0.246], requires_grad=True)
>>> qnode_conditional_op_on_zero(*pars)
tensor([0.88660045, 0.11339955], requires_grad=True)

For more examples on applying quantum functions conditionally, refer to the cond() documentation.

Postselecting mid-circuit measurements

PennyLane also supports postselecting on mid-circuit measurement outcomes by specifying the postselect keyword argument of measure(). Postselection discards outcomes that do not meet the criteria provided by the postselect argument. For example, specifying postselect=1 on wire 0 would be equivalent to projecting the state vector onto the \(|1\rangle\) state on wire 0, i.e., disregarding all outcomes where \(|0\rangle\) is measured on wire 0:

dev = qml.device("default.qubit")

@qml.qnode(dev)
def func(x):
    qml.RX(x, wires=0)
    m0 = qml.measure(0, postselect=1)
    qml.cond(m0, qml.PauliX)(wires=1)
    return qml.sample(wires=1)

By postselecting on 1, we only consider the 1 measurement outcome on wire 0. So, the probability of measuring 1 on wire 1 after postselection should also be 1. Executing this QNode with 10 shots:

>>> func(np.pi / 2, shots=10)
array([1, 1, 1, 1, 1, 1, 1])

Note that only 7 samples are returned. This is because samples that do not meet the postselection criteria are discarded. To learn more about postselecting mid-circuit measurements, refer to the measure() documentation.

Note

Currently, postselection support is only available on DefaultQubit. Using postselection on other devices will raise an error.

Mid-circuit measurement statistics

Statistics can be collected on mid-circuit measurements along with terminal measurement statistics. Currently, qml.probs, qml.sample, qml.expval, qml.var, and qml.counts are supported, and can be requested along with other measurements. The devices that currently support collecting such statistics are DefaultQubit, DefaultMixed, and DefaultQubitLegacy.

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

@qml.qnode(dev)
def func(x, y):
    qml.RX(x, wires=0)
    m0 = qml.measure(0)
    qml.cond(m0, qml.RY)(y, wires=1)
    return qml.probs(wires=1), qml.probs(op=m0)

Executing this QNode:

>>> func(np.pi / 2, np.pi / 4)
(tensor([0.9267767, 0.0732233], requires_grad=True),
 tensor([0.5, 0.5], requires_grad=True))

Users can also collect statistics on mid-circuit measurements manipulated using arithmetic/boolean operators. This works for both unary and binary operators. To see a full list of supported operators, refer to the measure() documentation. An example for collecting such statistics is shown below:

import pennylane as qml

dev = qml.device("default.qubit")

@qml.qnode(dev)
def circuit(phi, theta):
    qml.RX(phi, wires=0)
    m0 = qml.measure(wires=0)
    qml.RY(theta, wires=1)
    m1 = qml.measure(wires=1)
    return qml.sample(~m0 - 2 * m1)

Executing this QNode:

>>> circuit(1.23, 4.56, shots=5)
array([-1, -2,  1, -1,  1])

Collecting statistics for mid-circuit measurements manipulated using arithmetic/boolean operators is supported with qml.expval, qml.var, qml.sample, and qml.counts.

Moreover, statistics for multiple mid-circuit measurements can be collected by passing lists of mid-circuit measurement values to the measurement process:

import pennylane as qml

dev = qml.device("default.qubit")

@qml.qnode(dev)
def circuit(phi, theta):
    qml.RX(phi, wires=0)
    m0 = qml.measure(wires=0)
    qml.RY(theta, wires=1)
    m1 = qml.measure(wires=1)
    return qml.sample([m0, m1])

Executing this QNode:

>>> circuit(1.23, 4.56, shots=5)
array([[0, 1],
       [1, 1],
       [0, 1],
       [0, 0],
       [1, 1]])

Collecting statistics for sequences of mid-circuit measurements is supported with qml.sample, qml.probs, and qml.counts.

Warning

When collecting statistics for a list of mid-circuit measurements, values manipulated using arithmetic operators should not be used as this behaviour is not supported.

Changing the number of shots

For hardware devices where the number of shots determines the accuracy of the expectation value and variance, as well as the number of samples returned, it can sometimes be convenient to execute the same QNode with differing number of shots.

For simulators like default.qubit, finite shots will be simulated if we set shots to a positive integer.

The shot number can be changed on the device itself, or temporarily altered by the shots keyword argument when executing the QNode:

dev = qml.device("default.qubit", wires=1, shots=10)

@qml.qnode(dev)
def circuit(x, y):
    qml.RX(x, wires=0)
    qml.RY(y, wires=0)
    return qml.expval(qml.PauliZ(0))

# execute the QNode using 10 shots
result = circuit(0.54, 0.1)

# execute the QNode again, now using 1 shot
result = circuit(0.54, 0.1, shots=1)

With an increasing number of shots, the average over measurement samples converges to the exact expectation of an observable. Consider the following circuit:

# fix seed to make results reproducable
np.random.seed(1)

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

@qml.qnode(dev)
def circuit():
    qml.Hadamard(wires=0)
    return qml.expval(qml.PauliZ(0))

Running the simulator with shots=None returns the exact expectation.

>>> circuit(shots=None)
0.0

Now we set the device to return stochastic results, and increase the number of shots starting from 10.

>>> circuit(shots=10)
0.2
>>> circuit(shots=1000)
-0.062
>>> circuit(shots=100000)
0.00056

The result converges to the exact expectation.