tree-lstm

tree-lstm copied to clipboard

Hi @inyukwo1 , Thanks for the package. Just wondering whether this line should be ` cur_batch.batch_dgl_graph.ndata['x'])` instead as it would probably make more sense to feed the hidden state of...

maclandrol

Reduce func in NaryTreeLSTMCell

2

Hi, I notice that in your implementation of the NaryTreeLSTMCell, the reduce function returns `return {'iou': nodes.data['iou'] + self.U_iou(h_cat), 'c': c}` while implementations in DGL returns only `return {'iou': self.U_iou(h_cat),...

haorannlp