qml.capture.primitives.for_loop_prim

for_loop_prim = for_loop

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

code/api/pennylane.capture.primitives.for_loop_prim

 

Download Python script

 

Download Notebook

 

View on GitHub