然后,我们定义一个MLP模型类,继承自PyTorch的nn.Module类: classMLP(nn.Module):def__init__(self,input_dim,hidden_dim,output_dim):super(MLP,self).__init__()self.fc1=nn.Linear(input_dim,hidden_dim)self.fc2=nn.Linear(hidden_dim,output_dim)self.relu=nn.ReLU()self.softmax=nn.Softmax(dim...
您可以在torch_geometric中看到GINConv和GCNConvAPI。