* fix no_grad context error in dy2stat

* remove useless comments

* fix error by drop_kids in python

* add test and fix review