EGNN-PyTorch：基于PyTorch的E(n)等价图神经网络实现
5星
- 浏览量: 0
- 大小:None
- 文件类型：None
简介：
简介：EGNN-PyTorch是一款采用PyTorch框架开发的库，专门用于实现具有平移和旋转不变性的E(n)等价图神经网络，适用于物理系统、分子动力学及材料科学等领域。
EGNN-Pytorch（WIP）中的实现最终可用于Alphafold2的复制。安装方法为：`pip install egnn-pytorch`
用法示例：
```python
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)
```
带边的用法：
```python
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)
```
全部评论 (0)
还没有任何评论哟~
