[BugFix] hetero ogbn-mag full neighbor inference. (#4010)

* fix hetero ogbn-mag full neighbor inference.

* fix hetero
Co-authored-by: Mufei Li <mufeili1996@gmail.com>

examples/pytorch/rgcn-hetero-ogbn-mag/model.py

@@ -321,7 +321,7 @@ class EntityClassify(nn.Module):
                 if i < self._num_layers - 1:
                     h = self._apply_layers(i, h)
 
-                for ntype in h:
+                for ntype in out_nodes:
                     y[ntype][out_nodes[ntype]] = h[ntype]
 
             x = y