Encoder for unitaries

Model definition

---

source

Unitary_encoder_config

 Unitary_encoder_config (cond_emb_size:int, model_features:list[int],
                         num_heads:int, transformer_depths:list[int],
                         dropout:float)

---

source

Unitary_encoder

 Unitary_encoder (cond_emb_size, model_features=None, num_heads=8,
                  transformer_depths=[4, 4], dropout=0.1)

Encoder for unitary conditions.