Flatten layer of PyTorch build by sequential container

Answered here.