pygcn
pygcn copied to clipboard
Xavier initialization use output size instead of input size
I notice the standard deviation for initialization used output size instead of input size. Is this implementation intended?
self.weight = Parameter(torch.FloatTensor(in_features, out_features))
stdv = 1. / math.sqrt(self.weight.size(1))