"n">metric</span><span class="o">=</span><span class="kc">None</span><span class="p">):</span>
        <span class="sd">"""</span>
<span class="sd">        Compute the node representations and the graph representation(s).</span>

<span class="sd">        Parameters:</span>
<span class="sd">            graph (Graph): :math:`n` graph(s)</span>
<span class="sd">            input (Tensor): input node representations</span>
<span class="sd">            all_loss (Tensor, optional): if specified, add loss to this 