Pytorch flatten layer example. In this post, we extend that work by add...