【文件属性】：
文件名称：egnn-pytorch：在Pytorch中实现E（n）-等价图神经网络
文件大小：102KB
文件格式：ZIP
更新时间：2021-02-27 05:51:41
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)