378c264561d232dfe00d934409b64ab626a4915a,examples/pytorch/gcn/gcn.py,NodeApplyModule,forward,#NodeApplyModule#Any#,29

Before Change

        self.activation = activation

    def forward(self, nodes):
        h = self.linear(nodes.data["h"])
        if self.activation:
            h = self.activation(h)
        return {"h" : h}

After Change

    def forward(self, nodes):
        // normalization by square root of dst degree
        h = nodes.data["h"] * nodes.data["norm"]
        h = self.linear(h)
        if self.activation:
            h = self.activation(h)
        return {"h" : h}

Instances