esentations</span>
<span class="sd">            all_loss (Tensor, optional): if specified, add loss to this tensor</span>
<span class="sd">            metric (dict, optional): if specified, output metrics to this dict</span>

<span class="sd">        Returns:</span>
<span class="sd">            dict with ``residue_feature`` and ``graph_feature`` fields:</span>
<span class="sd">                residue representations of shape :math:`(|V_{res}|, d)`, graph representations of shape :math:`(n, d)`</span>
<span 