Make Conv2D to Vector conversions independent of axes