fix flatten layer backwards to dummy return