I tried to to get the input image with one dimension but I became an error every time, how I can use you code with one dimension?