Source code for pennylane.liealg.structure_constants
# Copyright 2025 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."""A function to compute the adjoint representation of a Lie algebra"""fromitertoolsimportcombinations,combinations_with_replacementfromtypingimportUnionimportnumpyasnpimportpennylaneasqmlfrompennylane.operationimportOperatorfrompennylane.pauliimportPauliSentence,PauliWordfrompennylane.typingimportTensorLikedef_all_commutators(ops):commutators={}for(j,op1),(k,op2)incombinations(enumerate(ops),r=2):res=op1.commutator(op2)ifres!=PauliSentence({}):commutators[(j,k)]=resreturncommutators
[docs]defstructure_constants(g:list[Union[Operator,PauliWord,PauliSentence]],pauli:bool=False,matrix:bool=False,is_orthogonal:bool=True,)->TensorLike:r""" Compute the structure constants that make up the adjoint representation of a Lie algebra. Given a DLA :math:`\{iG_1, iG_2, .. iG_d \}` of dimension :math:`d`, the structure constants yield the decomposition of all commutators in terms of DLA elements, .. math:: [i G_\alpha, i G_\beta] = \sum_{\gamma = 0}^{d-1} f^\gamma_{\alpha, \beta} iG_\gamma. The adjoint representation :math:`\left(\text{ad}(iG_\gamma)\right)_{\alpha, \beta} = f^\gamma_{\alpha, \beta}` is given by those structure constants, which can be computed via .. math:: f^\gamma_{\alpha, \beta} = \frac{\text{tr}\left(i G_\gamma \cdot \left[i G_\alpha, i G_\beta \right] \right)}{\text{tr}\left( iG_\gamma iG_\gamma \right)}. The inputs are assumed to be orthogonal unless ``is_orthogonal`` is set to ``False``. However, we neither assume nor enforce normalization of the DLA elements :math:`G_\alpha`. Args: g (List[Union[Operator, PauliWord, PauliSentence]]): The (dynamical) Lie algebra for which we want to compute the adjoint representation. DLAs can be generated by a set of generators via :func:`~lie_closure`. pauli (bool): Indicates whether it is assumed that :class:`~.PauliSentence` or :class:`~.PauliWord` instances are input. This can help with performance to avoid unnecessary conversions to :class:`~pennylane.operation.Operator` and vice versa. Default is ``False``. matrix (bool): Whether or not matrix matrix representations are used and output in the structure constants computation. Default is ``False``. is_orthogonal (bool): Whether the set of operators in ``g`` is orthogonal with respect to the trace inner product. Default is ``True``. Returns: TensorLike: The adjoint representation of shape ``(d, d, d)``, corresponding to indices ``(gamma, alpha, beta)``. .. seealso:: :func:`~lie_closure`, :func:`~center`, :class:`~pennylane.pauli.PauliVSpace`, `Demo: Introduction to Dynamical Lie Algebras for quantum practitioners <https://pennylane.ai/qml/demos/tutorial_liealgebra/>`__ **Example** Let us first generate the DLA of the transverse field Ising model using :func:`~lie_closure`. >>> n = 2 >>> gens = [X(i) @ X(i+1) for i in range(n-1)] >>> gens += [Z(i) for i in range(n)] >>> dla = qml.lie_closure(gens) >>> print(dla) [X(0) @ X(1), Z(0), Z(1), -1.0 * (Y(0) @ X(1)), -1.0 * (X(0) @ Y(1)), -1.0 * (Y(0) @ Y(1))] The dimension of the DLA is :math:`d = 6`. Hence, the structure constants have shape ``(6, 6, 6)``. >>> adjoint_rep = qml.structure_constants(dla) >>> adjoint_rep.shape (6, 6, 6) The structure constants tell us the commutation relation between operators in the DLA via .. math:: [i G_\alpha, i G_\beta] = \sum_{\gamma = 0}^{d-1} f^\gamma_{\alpha, \beta} iG_\gamma. Let us confirm those with an example. Take :math:`[iG_1, iG_3] = [iZ_0, -iY_0 X_1] = -i 2 X_0 X_1 = -i 2 G_0`, so we should have :math:`f^0_{1, 3} = -2`, which is indeed the case. >>> adjoint_rep[0, 1, 3] -2.0 We can also look at the overall adjoint action of the first element :math:`G_0 = X_{0} \otimes X_{1}` of the DLA on other elements. In particular, at :math:`\left(\text{ad}(iG_0)\right)_{\alpha, \beta} = f^0_{\alpha, \beta}`, which corresponds to the following matrix. >>> adjoint_rep[0] array([[ 0., 0., 0., 0., 0., 0.], [-0., 0., 0., -2., 0., 0.], [-0., 0., 0., 0., -2., 0.], [-0., 2., -0., 0., 0., 0.], [-0., -0., 2., 0., 0., 0.], [ 0., -0., -0., -0., -0., 0.]]) Note that we neither enforce nor assume normalization by default. To compute the structure constants of a non-orthogonal set of operators, use the option ``is_orthogonal=False``: >>> dla = [qml.X(0), qml.Y(0), qml.X(0) - qml.Z(0)] >>> adjoint_rep = qml.structure_constants(dla, is_orthogonal=False) >>> adjoint_rep[:, 0, 1] # commutator of X_0 and Y_0 consists of first and last operator array([-2., 0., 2.]) We can also use matrix representations for the computation, which is sometimes faster, in particular for sums of many Pauli words. This alters how the structure constants are computed internally, but it does not change the result. >>> adjoint_rep2 = qml.structure_constants(dla, is_orthogonal=False, matrix=True) >>> qml.math.allclose(adjoint_rep, adjoint_rep2) True We can also input the DLA in form of matrices. For that we use :func:`~lie_closure` with the ``matrix=True``. >>> n = 4 >>> gens = [qml.X(i) @ qml.X(i+1) + qml.Y(i) @ qml.Y(i+1) + qml.Z(i) @ qml.Z(i+1) for i in range(n-1)] >>> g = qml.lie_closure(gens, matrix=True) >>> g.shape (12, 16, 16) The DLA is represented by a collection of twelve :math:`2^4 \times 2^4` matrices. Hence, the dimension of the DLA is :math:`d = 12` and the structure constants have shape ``(12, 12, 12)``. >>> from pennylane.labs.dla import structure_constants_matrix >>> adj = structure_constants_matrix(g) >>> adj.shape (12, 12, 12) .. details:: :title: Mathematical details Consider a (dynamical) Lie algebra :math:`\{iG_1, iG_2, .. iG_d \}` of dimension :math:`d`. The defining property of the structure constants is that they express the decomposition of commutators in terms of the DLA elements, as described at the top. This can be written as .. math:: [i G_\alpha, i G_\beta] = \sum_{\gamma = 0}^{d-1} f^\gamma_{\alpha, \beta} iG_\gamma. Now we may multiply this equation with the adjoint of a DLA element and apply the trace: .. math:: \text{tr}\left(-i G_\eta \cdot \left[i G_\alpha, i G_\beta \right] \right) &= \text{tr}\left(-i G_\eta \sum_{\gamma = 0}^{d-1} f^\gamma_{\alpha, \beta} iG_\gamma\right)\\ &= \sum_{\gamma = 0}^{d-1} \underset{g_{\eta \gamma}}{\underbrace{ \text{tr}\left(-i G_\eta iG_\gamma\right)}} f^\gamma_{\alpha, \beta} \\ \Rightarrow\ f^\gamma_{\alpha, \beta} &= (g^{-1})_{\gamma \eta} \text{tr}\left(-i G_\eta \cdot \left[i G_\alpha, i G_\beta \right] \right). Here we introduced the Gram matrix :math:`g_{\alpha\beta} = \text{tr}(-iG_\alpha i G_\beta)` of the DLA elements. Note that this is just the projection of the commutator on the DLA element :math:`iG_\gamma` via the trace inner product. Now, if the DLA elements are orthogonal, as assumed by ``structure_constants`` by default, the Gram matrix will be diagonal and simply consist of some rescaling factors, so that the above computation becomes the equation from the very top: .. math:: f^\gamma_{\alpha, \beta} = \frac{\text{tr}\left(i G_\gamma \cdot \left[i G_\alpha, i G_\beta \right] \right)} {\text{tr}\left( iG_\gamma iG_\gamma \right)}. This is cheaper than computing the full Gram matrix, inverting it, and multiplying it to the trace inner products. For the case of an orthonormal set of operators, we even have :math:`g_{\alpha\beta}=\delta_{\alpha\beta}`, so that the division in this calculation can be skipped. """ifmatrix:return_structure_constants_matrix(g,is_orthogonal)ifany((getattr(op,"pauli_rep",None)isNone)foroping):raiseValueError(f"Cannot compute adjoint representation of non-pauli operators. Received {g}. If you want to use matrices, use structure_constants(.., matrix=True)")ifnotpauli:g=[op.pauli_repforoping]commutators=_all_commutators(g)rep=np.zeros((len(g),len(g),len(g)),dtype=float)fori,opinenumerate(g):# if is_orthogonal is activated we will use the norm_squared of the op, otherwise we won'tnorm_squared=(op@op).trace()ifis_orthogonalelse1for(j,k),resincommutators.items():# if is_orthogonal is activated, use v = ∑ (v · e_j / ||e_j||^2) * e_jvalue=(1j*(op@res).trace()).real/norm_squaredrep[i,j,k]=valuerep[i,k,j]=-valueifnotis_orthogonal:gram=np.zeros((len(g),len(g)),dtype=float)for(i,op1),(j,op2)incombinations_with_replacement(enumerate(g),r=2):gram[i,j]=gram[j,i]=(op1@op2).trace()# Contract the structure constants on the upper index with the Gram matrix, see derivationrep=np.tensordot(np.linalg.pinv(gram),rep,axes=[[-1],[0]])returnrep
def_structure_constants_matrix(g:TensorLike,is_orthogonal:bool=True)->TensorLike:r""" Compute the structure constants that make up the adjoint representation of a Lie algebra. This function computes the structure constants of a Lie algebra provided by their matrix matrix representation, obtained from, e.g., :func:`~lie_closure`. This is sometimes more efficient than using the sparse Pauli representations of :class:`~PauliWord` and :class:`~PauliSentence` that are employed in :func:`~structure_constants`, e.g., when there are few generators that are sums of many Paulis. .. seealso:: For details on the mathematical definitions, see :func:`~structure_constants` and the section "Lie algebra basics" in our `g-sim demo <https://pennylane.ai/qml/demos/tutorial_liesim/#lie-algebra-basics>`__. Args: g (np.array): The (dynamical) Lie algebra provided as matrix matrices, as generated from :func:`~lie_closure`. ``g`` should have shape ``(d, 2**n, 2**n)`` where ``d`` is the dimension of the algebra and ``n`` is the number of qubits. Each matrix ``g[i]`` should be Hermitian. is_orthogonal (bool): Whether or not the matrices in ``g`` are orthogonal with respect to the Hilbert-Schmidt inner product on (skew-)Hermitian matrices. If the inputs are orthogonal, it is recommended to set ``is_orthogonal`` to ``True`` to reduce computational cost. Defaults to ``True``. Returns: TensorLike: The adjoint representation of shape ``(d, d, d)``, corresponding to the indices ``(gamma, alpha, beta)``. **Example** Let us generate the DLA of the transverse field Ising model using :func:`~lie_closure`. >>> n = 4 >>> gens = [qml.X(i) @ qml.X(i+1) + qml.Y(i) @ qml.Y(i+1) + qml.Z(i) @ qml.Z(i+1) for i in range(n-1)] >>> g = qml.lie_closure(gens, matrix=True) >>> g.shape (12, 16, 16) The DLA is represented by a collection of twelve :math:`2^4 \times 2^4` matrices. Hence, the dimension of the DLA is :math:`d = 12` and the structure constants have shape ``(12, 12, 12)``. >>> from pennylane.labs.dla import structure_constants_matrix >>> adj = structure_constants_matrix(g) >>> adj.shape (12, 12, 12) **Internal representation** As mentioned above, the input is assumed to be a batch of Hermitian matrices, even though algebra elements are usually skew-Hermitian. That is, the input should represent the operators :math:`G_\alpha` for an algebra basis :math:`\{iG_\alpha\}_\alpha`. In an orthonormal basis of this form, the structure constants can then be computed simply via .. math:: f^\gamma_{\alpha, \beta} = \text{tr}[-i G_\gamma[iG_\alpha, iG_\beta]] = i\text{tr}[G_\gamma [G_\alpha, G_\beta]] \in \mathbb{R}. Possible deviations of an orthogonal basis from normalization is taken into account in a reduced version of the step for non-orthogonal bases below. **Structure constants in non-orthogonal bases** Structure constants are often discussed using an orthogonal basis of the algebra. This function can deal with non-orthogonal bases as well. For this, the Gram matrix :math:`g` between the basis elements is taken into account when computing the overlap of a commutator :math:`[iG_\alpha, iG_\beta]` with all algebra elements :math:`iG_\gamma`. The resulting formula reads .. math:: f^\gamma_{\alpha, \beta} &= \sum_\eta g^{-1}_{\gamma\eta} i \text{tr}[G_\eta [G_\alpha, G_\beta]]\\ g_{\gamma \eta} &= \text{tr}[G_\gamma G_\eta] \quad(\in\mathbb{R}) Internally, the commutators are computed by evaluating all operator products and subtracting suitable pairs of products from each other. These products can be reused to evaluate the Gram matrix as well. For orthogonal but not normalized bases, a reduced version of this step is used, only computing (and inverting) the diagonal of the Gram matrix. """ifgetattr(g[0],"wires",False):# operator inputall_wires=qml.wires.Wires.all_wires([_.wiresfor_ing])n=len(all_wires)assertall_wires.toset()==set(range(n))g=qml.math.array([qml.matrix(op,wire_order=range(n))foroping],dtype=complex,like=g[0])chi=2**nassertnp.shape(g)==(len(g),chi,chi)interface=qml.math.get_interface(g[0])ifisinstance(g[0],TensorLike)andisinstance(g,(list,tuple)):# list of matricesg=qml.math.stack(g,like=interface)chi=qml.math.shape(g[0])[0]assertqml.math.shape(g)==(len(g),chi,chi)assertqml.math.allclose(qml.math.transpose(qml.math.conj(g),(0,2,1)),g),"Input matrices to structure_constants not Hermitian"# compute all commutators by computing all products first.# Axis ordering is (dimg, chi, _chi_) x (dimg, _chi_, chi) -> (dimg, chi, dimg, chi)prod=qml.math.tensordot(g,g,axes=[[2],[1]])# The commutators now are the difference of prod with itself, with dimg axes swappedall_coms=prod-qml.math.transpose(prod,(2,1,0,3))# project commutators on the basis of g, see docstring for details.# Axis ordering is (dimg, _chi_, *chi*) x (dimg, *chi*, dimg, _chi_) -> (dimg, dimg, dimg)# Normalize trace inner product by dimension chiadj=qml.math.real(1j*qml.math.tensordot(g/chi,all_coms,axes=[[1,2],[3,1]]))ifis_orthogonal:# Orthogonal but not normalized inputs. Need to correct by (diagonal) Gram matrixifinterface=="tensorflow":importkeras# pylint: disable=import-outside-toplevelpre_diag=keras.ops.diagonal(keras.ops.diagonal(prod,axis1=1,axis2=3),axis1=0,axis2=1)else:# offset, axis1, axis2 arguments are called differently in torch, use positional argumentspre_diag=qml.math.diagonal(qml.math.diagonal(prod,0,1,3),0,0,1)gram_diag=qml.math.real(qml.math.sum(pre_diag,axis=0))adj=(chi/gram_diag[:,None,None])*adjelse:# Non-orthogonal inputs. Need to correct by (full) Gram matrix# Compute the Gram matrix and apply its (pseudo-)inverse to the obtained projections.# See the docstring for details.# The Gram matrix is just one additional diagonal contraction of the ``prod`` tensor,# across the Hilbert space dimensions. (dimg, _chi_, dimg, _chi_) -> (dimg, dimg)# This contraction is missing the normalization factor 1/chi of the trace inner product.gram_inv=qml.math.linalg.pinv(qml.math.real(qml.math.sum(qml.math.diagonal(prod,axis1=1,axis2=3),axis=-1)))# Axis ordering for contraction with gamma axis of raw structure constants:# (dimg, _dimg_), (_dimg_, dimg, dimg) -> (dimg, dimg, dim)# Here we add the missing normalization factor of the trace inner product (after inversion)adj=qml.math.tensordot(gram_inv*chi,adj,axes=1)returnadj