* remove flatten in nn.py under fluid

* fix test case

* fix test case

* fix codestyle

* fix codestyle

* fix code style

* remove extra test case

* remove test case