Pytorch maxout.