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

1""" Generic gates. """ 

2 

3 

4from jaxquantum.core.operators import identity 

5from jaxquantum.circuits.gates import Gate 

6import jax.numpy as jnp 

7from jaxquantum import Qarray, tensor 

8 

9 

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

11 

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

13 

14 return Gate.create( 

15 Ns, 

16 name="Id", 

17 params={}, 

18 gen_U=lambda params: Is, 

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

20 ts=ts, 

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

22 num_modes=len(Ns), 

23 )