qml.capture.primitives.cond_prim

cond_prim = cond

A subclass to JAX’s Primitive that works like a Python function when evaluating JVPTracers.

code/api/pennylane.capture.primitives.cond_prim

 

Download Python script

 

Download Notebook

 

View on GitHub