><span class="o">.</span><span class="n">Set2Set</span><span class="p">(</span><span class="n">feature_dim</span><span class="p">,</span> <span class="n">num_step</span><span class="o">=</span><span class="n">num_s2s_step</span><span class="p">)</span>

<div class="viewcode-block" id="MessagePassingNeuralNetwork.forward"><a class="viewcode-back" href="../../../api/models.html#torchdrug.models.MessagePassingNeuralNetwork.forward">[docs]</a>    <span class="k">def</span> <span class="nf">forward</span><span c