 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 class="sd">        """</span>
        <span class="nb">input</span> <span class="o">=</span> <span class="n">graph</span><span class="o">.</span><span class="n">residue_feature</span><span class="o">.</span><span class="n">float</span><span class="p">()</span>
        <span class="nb">input</span> <