Skip to content

Commit 5fac9f7

Browse files
authored
Merge pull request #337 from FluxML/doc
Update doc
2 parents b28a09f + 4f2480e commit 5fac9f7

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

docs/src/manual/graph_conv.md

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,16 @@ Reference: [GAT2018](@cite)
9797

9898
## Graph Attentional Layer v2
9999

100+
```math
101+
\textbf{x}_i' = \alpha_{i,i} \Theta \textbf{x}_i + \sum_{j \in \mathcal{N}(i)} \alpha_{i,j} \Theta \textbf{x}_j
102+
```
103+
104+
where the attention coefficient ``\alpha_{i,j}`` can be calculated from
105+
106+
```math
107+
\alpha_{i,j} = \frac{exp(\textbf{a}^T LeakyReLU(\Theta [\textbf{x}_i || \textbf{x}_j]))}{\sum_{k \in \mathcal{N}(i) \cup \{i\}} exp(\textbf{a}^T LeakyReLU(\Theta [\textbf{x}_i || \textbf{x}_k]))}
108+
```
109+
100110

101111
```@docs
102112
GATv2Conv

0 commit comments

Comments
 (0)