Format change using Black
This commit is contained in:
tankya2
committed by
Alessandro Candido
Alessandro Candido
parent
bed3a50be5
commit
3b00c81330
@@ -1,49 +1,66 @@
|
||||
|
||||
import cupy as cp
|
||||
import numpy as np
|
||||
|
||||
EINSUM_SYMBOLS_BASE = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
||||
|
||||
class QiboCircuitToEinsum:
|
||||
def __init__(self, circuit, dtype='complex128'):
|
||||
|
||||
self.backend = cp
|
||||
self.dtype = getattr(self.backend, dtype)
|
||||
|
||||
self.input_tensor_counter = np. zeros((circuit.nqubits,))
|
||||
class QiboCircuitToEinsum:
|
||||
def __init__(self, circuit, dtype="complex128"):
|
||||
self.backend = cp
|
||||
self.dtype = getattr(self.backend, dtype)
|
||||
|
||||
self.input_tensor_counter = np.zeros((circuit.nqubits,))
|
||||
self.gates = []
|
||||
for gate in circuit.queue:
|
||||
targets = list(gate.target_qubits)
|
||||
for target in targets:
|
||||
self.input_tensor_counter[target] = self.input_tensor_counter[target] + 1
|
||||
self.input_tensor_counter[target] = (
|
||||
self.input_tensor_counter[target] + 1
|
||||
)
|
||||
controls = list(gate.control_qubits)
|
||||
for control in controls:
|
||||
self.input_tensor_counter[control] = self.input_tensor_counter[control] + 1
|
||||
self.input_tensor_counter[control] = (
|
||||
self.input_tensor_counter[control] + 1
|
||||
)
|
||||
gate_qubits = controls + targets
|
||||
self.gates.append((cp.asarray(gate.matrix).reshape((2,) * 2 * len(gate_qubits)), gate_qubits))
|
||||
self.gates.append(
|
||||
(
|
||||
cp.asarray(gate.matrix).reshape((2,) * 2 * len(gate_qubits)),
|
||||
gate_qubits,
|
||||
)
|
||||
)
|
||||
|
||||
self.qubit_name = [indx for indx, value in enumerate(self.input_tensor_counter) if value > 0]
|
||||
self.qubit_name = [
|
||||
indx for indx, value in enumerate(self.input_tensor_counter) if value > 0
|
||||
]
|
||||
|
||||
def state_vector(self):
|
||||
|
||||
input_tensor_count = np.count_nonzero(self.input_tensor_counter)
|
||||
|
||||
input_operands = self._get_bitstring_tensors('0'*input_tensor_count, self.dtype, backend=self.backend)
|
||||
|
||||
mode_labels, qubits_frontier, next_frontier = self._init_mode_labels_from_qubits(self.qubit_name)
|
||||
input_operands = self._get_bitstring_tensors(
|
||||
"0" * input_tensor_count, self.dtype, backend=self.backend
|
||||
)
|
||||
|
||||
gate_mode_labels, gate_operands = self._parse_gates_to_mode_labels_operands(self.gates,
|
||||
qubits_frontier,
|
||||
next_frontier)
|
||||
(
|
||||
mode_labels,
|
||||
qubits_frontier,
|
||||
next_frontier,
|
||||
) = self._init_mode_labels_from_qubits(self.qubit_name)
|
||||
|
||||
gate_mode_labels, gate_operands = self._parse_gates_to_mode_labels_operands(
|
||||
self.gates, qubits_frontier, next_frontier
|
||||
)
|
||||
|
||||
operands = input_operands + gate_operands
|
||||
mode_labels += gate_mode_labels
|
||||
|
||||
expression = self._convert_mode_labels_to_expression(mode_labels, qubits_frontier)
|
||||
expression = self._convert_mode_labels_to_expression(
|
||||
mode_labels, qubits_frontier
|
||||
)
|
||||
|
||||
return expression, operands
|
||||
|
||||
def _get_symbol(self,i):
|
||||
def _get_symbol(self, i):
|
||||
"""
|
||||
Return a Unicode as label for index.
|
||||
|
||||
@@ -53,33 +70,26 @@ class QiboCircuitToEinsum:
|
||||
return EINSUM_SYMBOLS_BASE[i]
|
||||
return chr(i + 140)
|
||||
|
||||
def _init_mode_labels_from_qubits(self,qubits):
|
||||
|
||||
frontier_dict ={}
|
||||
def _init_mode_labels_from_qubits(self, qubits):
|
||||
frontier_dict = {}
|
||||
n = len(qubits)
|
||||
for x in range(n):
|
||||
frontier_dict[qubits[x]]=x
|
||||
frontier_dict[qubits[x]] = x
|
||||
return [[i] for i in range(n)], frontier_dict, n
|
||||
|
||||
def _get_bitstring_tensors(self, bitstring, dtype=np.complex128, backend=cp):
|
||||
|
||||
asarray = backend.asarray #_get_backend_asarray_func(backend)
|
||||
asarray = backend.asarray # _get_backend_asarray_func(backend)
|
||||
state_0 = asarray([1, 0], dtype=dtype)
|
||||
state_1 = asarray([0, 1], dtype=dtype)
|
||||
|
||||
basis_map = {'0': state_0,
|
||||
'1': state_1}
|
||||
|
||||
basis_map = {"0": state_0, "1": state_1}
|
||||
|
||||
operands = [basis_map[ibit] for ibit in bitstring]
|
||||
return operands
|
||||
|
||||
def _parse_gates_to_mode_labels_operands(
|
||||
self,
|
||||
gates,
|
||||
qubits_frontier,
|
||||
next_frontier
|
||||
self, gates, qubits_frontier, next_frontier
|
||||
):
|
||||
|
||||
mode_labels = []
|
||||
operands = []
|
||||
|
||||
@@ -92,16 +102,19 @@ class QiboCircuitToEinsum:
|
||||
output_mode_labels.append(next_frontier)
|
||||
qubits_frontier[q] = next_frontier
|
||||
next_frontier += 1
|
||||
mode_labels.append(output_mode_labels+input_mode_labels)
|
||||
mode_labels.append(output_mode_labels + input_mode_labels)
|
||||
return mode_labels, operands
|
||||
|
||||
def _convert_mode_labels_to_expression(self,input_mode_labels, output_mode_labels):
|
||||
|
||||
def _convert_mode_labels_to_expression(self, input_mode_labels, output_mode_labels):
|
||||
out_list = []
|
||||
for key in output_mode_labels:
|
||||
out_list.append(output_mode_labels[key])
|
||||
|
||||
input_symbols = [''.join(map(self._get_symbol, idx)) for idx in input_mode_labels]
|
||||
expression = ','.join(input_symbols) + '->' + ''.join(map(self._get_symbol, out_list))
|
||||
input_symbols = [
|
||||
"".join(map(self._get_symbol, idx)) for idx in input_mode_labels
|
||||
]
|
||||
expression = (
|
||||
",".join(input_symbols) + "->" + "".join(map(self._get_symbol, out_list))
|
||||
)
|
||||
|
||||
return expression
|
||||
|
||||
Reference in New Issue
Block a user