Template Class AdjointJacobian

• Defined in File AdjointJacobianKokkos.hpp

Inheritance Relationships

Base Type

• public AdjointJacobianBase< StateVectorT, AdjointJacobian< StateVectorT > >

Class Documentation

template<class StateVectorT>
class AdjointJacobian : public AdjointJacobianBase<StateVectorT, AdjointJacobian<StateVectorT>>

Kokkos-enabled adjoint Jacobian evaluator following the method of arXiv:2009.02823.

Template Parameters

StateVectorT – State vector type.

Public Functions

AdjointJacobian() = default

inline void adjointJacobian(std::span<PrecisionT> jac, const JacobianData<StateVectorT> &jd, const StateVectorT &ref_data, bool apply_operations = false)

Calculates the Jacobian for the statevector for the selected set of parametric gates.

For the statevector data associated with psi of length num_elements, we make internal copies, one per required observable. The operations will be applied to the internal statevector copies, with the operation indices participating in the gradient calculations given in trainableParams, and the overall number of parameters for the gradient calculation provided within num_params. The resulting row-major ordered jac matrix representation will be of size jd.getSizeStateVec() * jd.getObservables().size(). OpenMP is used to enable independent operations to be offloaded to threads.

Note

Only gates with pre-defined generators can be differentiated. For example, QubitUnitary is not differentiable as there is no generator defined for this gate.

Parameters

• jac – Preallocated vector for Jacobian data results.

• jd – JacobianData represents the QuantumTape to differentiate.

• apply_operations – Indicate whether to apply operations to tape.psi prior to calculation.

api/classPennylane_1_1LightningKokkos_1_1Algorithms_1_1AdjointJacobian

 

Download Python script

 

Download Notebook

 

View on GitHub