qml.capture.primitives.for_loop_prim

for_loop_prim = for_loop

A subclass for JAX’s Primitive that differentiates between different classes of primitives and automatically makes parameters hashable for JAX 0.7.0+.

code/api/pennylane.capture.primitives.for_loop_prim

 

Download Python script

 

Download Notebook

 

View on GitHub