Batch support for TreeLSTM

[jinfengr]

Existing implementation doesn't support forward/backward with batch of trees as inputs, which is slow in training and inference. The pull requests support batch operation for TreeLSTM, and reproduces the exact same results as without batch.

To run with batch:

- python main.py --use_batch