tomodrgn.set_transformer

Set transformer classes adapted from juho-lee/set_transformer under MIT license

Classes

MAB	Multihead Attention module.
PMA	Pooled MultiHead Attention module.
SAB	Set Attention module.
SetTransformer	Set Transformer module, the primary contribution of juho-lee/set_transformer.