"dgl_sparse/git@developer.sourcefind.cn:OpenDAS/dgl.git" did not exist on "bb1f8850e5832309e910e967871cf099f83b37e9"
Unverified Commit 43b9d40e authored by Mufei Li's avatar Mufei Li Committed by GitHub
Browse files

[Doc] Fix Doc for EGNNConv (#3919)


Co-authored-by: default avatarMinjie Wang <wmjlyjemaine@gmail.com>
parent 01e50626
......@@ -10,9 +10,13 @@ class EGNNConv(nn.Module):
Neural Networks <https://arxiv.org/abs/2102.09844>`__
.. math::
m_{ij}=\phi_e(h_i^l, h_j^l, ||x_i^l-x_j^l||^2, a_{ij})
x_i^{l+1} = x_i^l + C\sum_{j\in\mathcal{N}(i)}(x_i^l-x_j^l)\phi_x(m_{ij})
m_i = \sum_{j\in\mathcal{N}(i)} m_{ij}
h_i^{l+1} = \phi_h(h_i^l, m_i)
where :math:`h_i`, :math:`x_i`, :math:`a_{ij}` are node features, coordinate
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment