dq.to_jax

to_jax(x: QArrayLike) -> Array

Convert a qarray-like into a JAX array.

Parameters

• x –

  Object to convert.

Returns

JAX array.

Examples

>>> dq.to_jax(dq.fock(3, 1))
Array([[0.+0.j],
       [1.+0.j],
       [0.+0.j]], dtype=complex64)
>>> dq.to_jax([qt.sigmax(), qt.sigmay(), qt.sigmaz()])
Array([[[ 0.+0.j,  1.+0.j],
        [ 1.+0.j,  0.+0.j]],

       [[ 0.+0.j,  0.-1.j],
        [ 0.+1.j,  0.+0.j]],

       [[ 1.+0.j,  0.+0.j],
        [ 0.+0.j, -1.+0.j]]], dtype=complex64)