文件名称：lie-transformer-pytorch:在Pytorch中实现等变自注意的Lie变压器
文件大小：181KB
文件格式：ZIP
更新时间：2024-05-29 23:16:03
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)