_loss</span><span class="o">=</span><span class="kc">None</span><span class="p">,</span> <span class="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 residue representations and the graph representation(s).</span>

<span class="sd">        Parameters:</span>
<span class="sd">            graph (Protein): :math:`n` protein(s)</span>
<span class="sd">            input (Tensor): input node repr