Encoder for unitaries

Model definition

---

source

Unitary_encoder_config

def Unitary_encoder_config(
    cond_emb_size:int, model_features:list, num_heads:int, transformer_depths:list, dropout:float
)->None:

---

source

Unitary_encoder

def Unitary_encoder(
    cond_emb_size, model_features:NoneType=None, num_heads:int=8, transformer_depths:list=[4, 4], dropout:float=0.1
):

Encoder for unitary conditions.