Change the input dimension of conv and convExpend to flat array