hqm.layers.basiclayer

 1import sys
 2
 3sys.path += ['.', './utils/', '/circuits/']
 4
 5from hqm.circuits.circuit import QuantumCircuit
 6from hqm.utils.aiinterface import AIInterface
 7
 8
 9class BasicLayer:
10    '''
11        Basic Quantum Layer
12    '''
13
14    def __init__(self, qcircuit : QuantumCircuit, aiframework : str) -> None:
15        '''
16        BasicLayer constructor.  
17
18        Parameters:  
19        -----------  
20        - qcircuit : hqm.circuits.circuit.QuantumCircuit  
21            QuantumCircuit object to be embedded into the quantum layer
22        - aiframework : str    
23            string representing the AI framework in use, can be 'torch' or 'keras'. This will create  
24            a compatible trainable layer for the framework.   
25
26        Returns:    
27        --------     
28        Nothing, a BasicLayer object will be created.  
29        '''
30        
31        if aiframework not in ['torch', 'keras']: raise Exception(f"Accepted values for framerwork are 'torch' or 'keras', found {aiframework}")
32
33        self.aiframework = aiframework
34        self.n_qubits    = qcircuit.n_qubits
35        self.qlayer      = AIInterface.network_layer(
36                                circuit      = qcircuit.circuit, 
37                                weight_shape = qcircuit.weight_shape, 
38                                n_qubits     = qcircuit.n_qubits, 
39                                aiframework  = self.aiframework
40                            )