lucidrains/lie-transformer-pytorch

Implementation of Lie Transformer, Equivariant Self-Attention, in Pytorch

语言：Python · GitHub 星标：98 · Fork：10

话题：artificial-intelligence · attention-mechanism · deep-learning · equivariance · se3 · transformer

在 GitHub 上查看原仓库