https://github.com/dasguptar/treelstm.pytorch/blob/228a314add09fc7f39ea752aa7b1fcf756cfe277/treelstm/dataset.py#L63