Method forward

docs/source/tutorial_cn/examples/4custom_gnn.py:59–62

(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

# %%
# 将自定义的GNN模型与Cogdl一起使用