from below codw when 1 image is passed the shape i get is (1,3,640,640).

```
def shape(im1):
    image_src = Image.open(im1)
    print('Loaded Image Info : ',image_src.format, image_src.size, image_src.mode) # size order : width*height

    # Resize to img_size_w, img_size_h
    resized =  image_src.resize((640, 640))   # To be imblemnated : letterbox_image(image_src, (img_size_w, img_size_h))
    print('After resizing :' ,resized.size, resized.mode) # size order : width*height
    #display(resized)

    # Preprocess the image
    img_in = np.transpose(resized, (2, 0, 1)).astype(np.float32)  # HWC -> CHW
    img_in = np.expand_dims(img_in, axis=0) # Add redundant dimension for batch-size (Assumed to be 1, check batch_size = session.get_inputs()[0].shape[0])
    img_in /= 255.0 # Normalize all pixels
    print('Batch-Size, Channel, Height, Width : ',img_in.shape)
    return img_in

```

What to make changes in the code so that if 2 images are passed they stack together and give shape as (2,3,640,640) .