# (x_patches, 128)
pooled_by_rows = image_patch_embeddings.mean(dim=0)