【文件属性】：
文件名称：lie-transformer-pytorch:在Pytorch中实现等变自注意的Lie变压器
文件大小：181KB
文件格式：ZIP
更新时间：2021-05-10 05:29:23
deep-learning artificial-intelligence transformer attention-mechanism se3
谎言变压器-火炬
在Pytorch中实现等距自注意的。 此存储库中仅存在SE3版本，因为Alphafold2复制可能需要它。
安装
$ pip install lie-transformer-pytorch
用法
import torch
from lie_transformer_pytorch import LieTransformer
model = LieTransformer (
dim = 512 ,
depth = 2 ,
heads = 8 ,
dim_head = 64 ,
liftsamples = 4
)
coors = torch . randn ( 1 , 64 , 3 )
features = torch . randn ( 1 , 64 , 512 )
mask = torch . ones ( 1 , 64 ). boo
【文件预览】：
lie-transformer-pytorch-main
----lie_transformer_pytorch()
--------reversible.py(5KB)
--------se3.py(7KB)
--------lie_transformer_pytorch.py(15KB)
--------__init__.py(75B)
----.github()
--------workflows()
----LICENSE(1KB)
----setup.cfg(81B)
----setup.py(882B)
----.gitignore(2KB)
----diagram.png(116KB)
----diagram-2.png(53KB)
----tests.py(369B)
----README.md(3KB)