# (1030, 128)
image_embedding = image_embeddings[0] # take the first element of the batch