i made a neural network with a (20,20) grid (shape = (20,20,1)) input and a vector of size 4 (shape = (4)). i want to use it on a sample of size 1. So if X1 is a (20,20,1