Skip to content

Commit

Permalink
Update gatv2conv.py doc (dmlc#3486)
Browse files Browse the repository at this point in the history
  • Loading branch information
shakedbr authored Nov 9, 2021
1 parent fe6e01a commit db78fac
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions python/dgl/nn/pytorch/conv/gatv2conv.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,16 +20,16 @@ class GATv2Conv(nn.Module):
over an input signal.
.. math::
h_i^{(l+1)} = \sum_{j\in \mathcal{N}(i)} \alpha_{i,j} W^{(l)}_{right} h_j^{(l)}
h_i^{(l+1)} = \sum_{j\in \mathcal{N}(i)} \alpha_{ij}^{(l)} W^{(l)}_{right} h_j^{(l)}
where :math:`\alpha_{ij}` is the attention score bewteen node :math:`i` and
node :math:`j`:
.. math::
\alpha_{ij}^{l} &= \mathrm{softmax_i} (e_{ij}^{l})
\alpha_{ij}^{(l)} &= \mathrm{softmax_i} (e_{ij}^{(l)})
e_{ij}^{l} &= \vec{a}^T\mathrm{LeakyReLU}\left(
W^{(l)}_{left} h_{i} + W^{(l)}_{right} h_{j}]\right)
e_{ij}^{(l)} &= \vec{a}^T^{(l)}\mathrm{LeakyReLU}\left(
W^{(l)}_{left} h_{i} + W^{(l)}_{right} h_{j}\right)
Parameters
----------
Expand Down

0 comments on commit db78fac

Please sign in to comment.