I am using pretrained VGG16 model to get features of an image. Image dimension is 256X340 however, VGG16 takes image of dimension 244X244, so I made 5 images out of original image, each of dimension 244X244 (topleft, topright, bottomleft, bottomright, center). Now I am passing these 5 images to model, dimension [5, 3, 244, 244] and getting output of [5, 4096] (modified last layer of VGG16 to get 4096 feats). Now I want to take the average of these 5 outputs so that I can get a final set of feature for original image.  Any suggestion on how to do this ?