The dataset is of images and the final output I am expecting is the feature extraction of those images. and I have not stored anything in target, because that is giving me error of singleton dimension.