Everything is okay until I convert my image to grayscale. So the rgb\'s shape is (256, 256, 3) but grayscale has (256, 256). When I feed it, I get that error.