Method forward

docs/source/examples/4custom_gnn.py:57–60

(self, graph, x)

        self.fc = torch.nn.Linear(in_feats, out_feats)

    def forward(self, graph, x):
        h = self.fc(x)
        h = spmm(graph, h)
        return h

# %%
# Use Custom models with CogDL