-\mathbf{q}^*_t = [\mathbf{q}; \mathbf{r}]\]</p><p>where <code>N</code> is the number of nodes in the graph, <code>LSTM</code> is a Long-Short-Term-Memory network with <code>n_layers</code> layers, input size <code>2*n_in</code> and output size <code>n_in</code>.</p><p>Given a batch of graphs <code>g</code> and node features <code>x</code>, the layer returns a matrix of size <code>(2*n_in, n_graphs)</code>. ```</p></div><a class="docs-sourcelink" target="_blank" href="https://github.com/JuliaGraphs/GraphNeuralNetworks.jl/blob/c9283d721e7fdc9378a8e13c3644b76787e2a5df/GraphNeuralNetworks/src/layers/pool.jl#L126-L143">source</a></section></article><article class="docstring"><header><a class="docstring-article-toggle-button fa-solid fa-chevron-down" href="javascript:;" title="Collapse docstring"></a><a class="docstring-binding" id="GraphNeuralNetworks.TopKPool" href="#GraphNeuralNetworks.TopKPool"><code>GraphNeuralNetworks.TopKPool</code></a> — <span class="docstring-category">Type</span><span class="is-flex-grow-1 docstring-article-toggle-button" title="Collapse docstring"></span></header><section><div><pre><code class="language-julia hljs">TopKPool(adj, k, in_channel)</code></pre><p>Top-k pooling layer.</p><p><strong>Arguments</strong></p><ul><li><code>adj</code>: Adjacency matrix of a graph.</li><li><code>k</code>: Top-k nodes are selected to pool together.</li><li><code>in_channel</code>: The dimension of input channel.</li></ul></div><a class="docs-sourcelink" target="_blank" href="https://github.com/JuliaGraphs/GraphNeuralNetworks.jl/blob/c9283d721e7fdc9378a8e13c3644b76787e2a5df/GraphNeuralNetworks/src/layers/pool.jl#L101-L111">source</a></section></article></article><nav class="docs-footer"><a class="docs-footer-prevpage" href="../conv/">« Convolutional layers</a><a class="docs-footer-nextpage" href="../temporalconv/">Temporal Convolutional layers »</a><div class="flexbox-break"></div><p class="footer-message">Powered by <a href="https://github.com/JuliaDocs/Documenter.jl">Documenter.jl</a> and the <a href="https://julialang.org/">Julia Programming Language</a>.</p></nav></div><div class="modal" id="documenter-settings"><div class="modal-background"></div><div class="modal-card"><header class="modal-card-head"><p class="modal-card-title">Settings</p><button class="delete"></button></header><section class="modal-card-body"><p><label class="label">Theme</label><div class="select"><select id="documenter-themepicker"><option value="auto">Automatic (OS)</option><option value="documenter-light">documenter-light</option><option value="documenter-dark">documenter-dark</option><option value="catppuccin-latte">catppuccin-latte</option><option value="catppuccin-frappe">catppuccin-frappe</option><option value="catppuccin-macchiato">catppuccin-macchiato</option><option value="catppuccin-mocha">catppuccin-mocha</option></select></div></p><hr/><p>This document was generated with <a href="https://github.com/JuliaDocs/Documenter.jl">Documenter.jl</a> version 1.7.0 on <span class="colophon-date" title="Monday 4 November 2024 12:34">Monday 4 November 2024</span>. Using Julia version 1.10.5.</p></section><footer class="modal-card-foot"></footer></div></div></div></body></html>
0 commit comments