文件名称：egnn-pytorch：在Pytorch中实现E（n）-等价图神经网络
文件大小：102KB
文件格式：ZIP
更新时间：2024-03-18 23:38:21
deep-learning artificial-intelligence graph-neural-network equivariance Python
EGNN-Pytorch（WIP） 中的实现。 最终可用于Alphafold2复制。 安装 $ pip install egnn-pytorch 用法 import torch from egnn_pytorch import EGNN layer1 = EGNN ( dim = 512 ) layer2 = EGNN ( dim = 512 ) feats = torch . randn ( 1 , 16 , 512 ) coors = torch . randn ( 1 , 16 , 3 ) feats , coors = layer1 ( feats , coors ) feats , coors = layer2 ( feats , coors ) # (1, 16, 512), (1, 16, 3) 带边 import torch from egnn_pytorch impo
【文件预览】：
egnn-pytorch-main
----.github()
--------workflows()
----LICENSE(1KB)
----egnn.png(98KB)
----egnn_pytorch()
--------egnn_pytorch.py(2KB)
--------__init__.py(43B)
----setup.py(768B)
----.gitignore(2KB)
----README.md(1KB)