Flatten add kernels dim into features

Flatten add kernels dim into features #506