Maxout now supports more than 2 dims