帮我写一段图神经网络gcn复现的实现代码

图神经网络GCN复现的实现代码：import torch
from torch import nn
from torch.nn import functional as Fclass GCN(nn.Module):
def __init__(self, in_features, out_features):
super(GCN, self).__init__()
self.in_features = in_features
self.out_features = out_features
self.weight = nn.Parameter(torch.FloatTensor(in_features, out_features))
self.bias = nn.Parameter(torch.FloatTensor(out_features))
self.reset_parameters() def reset_parameters(self):
stdv = 1. / math.sqrt(self.weight.size(1))
self.weight.data.uniform_(-stdv, stdv)
self.bias.data.uniform_(-stdv, stdv) def forward(self, input, adj):
support = torch.mm(input, self.weight)
output = torch.spmm(adj, support)
output = output + self.bias
return output