ValueError: could not broadcast input array from shape (1,2048) into shape (98,1024)

The code is as follow. Kindly give some suggestion on it. plz..

train_features=np.zeros((3056,98,1024))
test_features=np.zeros((764,98,1024))
for row in tqdm(range(train.shape[0])):
    image_1=train.iloc[row]["image1"]
    image_2=train.iloc[row]["image2"]
    train_features[row]=(image_feature_extraction(image_1,image_2))

below are image feature functions

def image_feature_extraction(image1,image2):
  
  image_1 = Image.open(image1)
  image_1= np.asarray(image_1.convert("RGB"))
  
  image_2=Image.open(image2)
  image_2 = np.asarray(image_2.convert("RGB"))

  image_1=image_1/255
  image_2=image_2/255

  image_1 = cv2.resize(image_1,(224,224))
  image_2 = cv2.resize(image_2,(224,224))
    
  image_1= np.expand_dims(image_1, axis=0)
  image_2= np.expand_dims(image_2, axis=0)
    
  image_1_out=final_chexnet_model(image_1)
  image_2_out=final_chexnet_model(image_2)
  
  conc=np.concatenate((image_1_out,image_2_out),axis=-1)
  
  image_feature=tf.reshape(conc, (conc.shape[0], -1, conc.shape[-1]))
  
  return image_feature