on(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 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 ``node_feature`` 