SE3变压器-Pytorch 在Pytorch中实现实现等价自注意。 复制Alphafold2结果和其他药物发现应用可能需要。 等方差示例 安装 $ pip install se3-transformer-pytorch 用法 import torch from se3_transformer_pytorch import SE3Transformer model = SE3Transformer ( dim = 512 , heads = 8 , depth = 6 , dim_head = 64 , num_degrees = 4 , valid_radius = 10 ) feats = torch . randn ( 1 , 1024 , 512 ) coors = torch . randn ( 1 , 1024 , 3 ) m