En-transformer：E（n）-Transformer的实现，将Welling的E（n）-等变图神经网络的思想扩展到关注

E（n）-等变变压器（wip） E（n）-等变变压器的实现，并引人关注地扩展了Welling的的思想。 安装 $ pip install En-transformer 用法 import torch from en_transformer import EnTransformer model = EnTransformer ( dim = 512 , depth = 4 , dim_head = 64 , heads = 8 , edge_dim = 4 , fourier_features = 2 ) feats = torch . randn ( 1 , 16 , 512 ) coors = torch . randn ( 1 , 16 , 3 ) edges = torch . randn ( 1 , 16 , 16 , 4 ) fe