Coverage for jaxquantum / circuits / library / generic.py: 67%

1""" Generic gates. """ 

2 

3 

4from jaxquantum.core.operators import identity 

5from jaxquantum.circuits.gates import Gate 

6from jaxquantum import Qarray, tensor 

7 

8 

9def Id(Ns, ts=None, c_ops=None): 

10 

11 Is = tensor(*[identity(N) for N in Ns]) 

12 

13 return Gate.create( 

14 Ns, 

15 name="Id", 

16 params={}, 

17 gen_U=lambda params: Is, 

18 gen_Ht=lambda params: (lambda t: 0*Is), 

19 ts=ts, 

20 gen_c_ops=lambda params: Qarray.from_list([]) if c_ops is None else c_ops, 

21 num_modes=len(Ns), 

22 )