lucidrains / isab-pytorch

An implementation of (Induced) Set Attention Block, from the Set Transformers paper
MIT License
56 stars 3 forks source link