a9ffb59e662398d9c40ed6ebe288e5542715e0dd,examples/pytorch/tree_lstm/tree_lstm.py,ChildSumTreeLSTMCell,reduce_func,#ChildSumTreeLSTMCell#Any#,28

Before Change

        // equation (4)
        wx = self.W_f(nodes.data["x"]).unsqueeze(1)  // shape: (B, 1, H)
        uh = self.U_f(nodes.mailbox["h"]) // shape: (B, deg, H)
        f = th.sigmoid(wx + uh)  // shape: (B, deg, H)
        // equation (7) second term
        c_tild = th.sum(f * nodes.mailbox["c"], 1)
        return {"h_tild" : h_tild, "c_tild" : c_tild}

After Change

    def reduce_func(self, nodes):
        h_cat = nodes.mailbox["h"].view(nodes.mailbox["h"].size(0), -1)
        f = th.sigmoid(self.U_f(h_cat)).view(*nodes.mailbox["h"].size())
        c = th.sum(f * nodes.mailbox["c"], 1)
        return {"iou": self.U_iou(h_cat), "c": c}

Instances