Can't measure the fidelity of the states.

[Shuhul24]

I have measure the quantum states using tfq.layers.State(). The output of tfq.layers.State() is of the form tf.RaggedTensor. Now I have measured two different states, using the same method. (say state1 and state2)
After using cirq.fidelity(state1, state2), I am getting the following error:

ValueError: Exception encountered when calling layer "generator__discriminator_6" (type Generator_Discriminator).

Failed to infer the qid shape of the given states. Please specify the qid shape explicitly using the `qid_shape` argument.

Call arguments received:
  • inputs=tf.Tensor(shape=(1,), dtype=string)

Why is it so? And how can I measure the fidelity of the states using tfq.layers.State()?

[lockwo]

Cirq doesn't play well with tensors, just convert it to a numpy array. E.g. this worked:

import tensorflow as tf
import tensorflow_quantum as tfq 
import numpy as np
import cirq
import sympy

qubits = [cirq.GridQubit(0, i) for i in range(4)]

params = sympy.symbols("q0:4")

c = cirq.Circuit()
for i in range(len(qubits)):
  c += cirq.ry(params[i]).on(qubits[i])

state_layer = tfq.layers.State()

state = state_layer(c, symbol_names=params, symbol_values=np.array([[1.0, 2.0, 3.0, 4.0]])).to_tensor()[0].numpy()

cirq.fidelity(state, state)

Also worth noting that TFQ has it's own mechanism for fidelity computations, so you don't have to compute the full state: https://www.tensorflow.org/quantum/api_docs/python/tfq/math/fidelity

[Shuhul24]

But the documentation says that it measure the fidelity between the "circuits". I don't quite get what does it mean to measure the fidelity between circuits instead of quantum states. (As it is pretty obvious from the definition of fidelity)

[lockwo]

It does measure the fidelity between states, but the input type is circuits and it uses that terminology to reflect that (since you could have many different parameters and stuff and it works with that). Since you don't input states to the function, you just work with circuits (that way you don't have to generate and store the states), it can be faster and easier (and works more smoothly with TFQ, since fidelity is a differentiable op).

[Shuhul24]

Is there any manner that I can measure the quantum relative entropy between two states? Or can I measure trace distance between two quantum states.
Also, I believe, the fidelity between pure and mixed states has some different notions as mixed states are ensembles of probabilities and states. Hence the code that you shared above is regardless of the fact the input quantum state is mixed or pure state?

PS: Quantum relative entropy, $S$ between two states $a$ and $b$ can be written as:
$$S = Tr(a(log(a) - log(b)))$$

[lockwo]

Yea cirq.fidelity says it takes statevectors or density matrices, so either should work. Regarding relative entropy, your best bet would probably be to take the output of the state layer and just compute it manually.

[Shuhul24]

What about trace distance? Can that be evaluated?

[lockwo]

Sure you could compute that as well. Just get the state and you can compute pretty much anything from it (of course none of this will be differentiable).

[Shuhul24]

But according to the formula of relative entropy I need to evaluate logarithm of a density matrix. How is that possible?

[lockwo]

Well you have a density matrix (either naturally or just convert the pure state to one, there is also maybe a simpler formula for pure states like there is for fidelity but I don't know it), and just do 'tf.linalg.logm'.

[lockwo]

Also for trace distance I thought you only needed the square root of a matrix?

[Shuhul24]

So how can I evaluate trace of a density matrix? cirq.partial_trace is something I don't think would work.

[lockwo]

You want to evaluate the trace of a matrix, just use tf.linalg.trace

[Shuhul24]

I think tf.linalg.logm aren't being applied on tf.RaggedTensor. Even after using .to_tensor() it says the tensor is not a square matrix. How to work around this?

[Shuhul24]

Is it because tfq.layers.State returns a non-square density matrix?

[lockwo]

Are you putting in a pure state? If so that's not a matrix and you need to convert it to a density matrix (of course the trace of a pure state density matrix is just 1)