Превращение Numpy массива символов c тензоров в пакетную форму

Question

# Modify and reshape the image batch
X_train = list(map(load_image_train, X_train[:100]))

X_train - это массив numpy тензоров символов c, где каждый элемент представляет собой (128, 256, 1) изображение в градациях серого. Это работает как задумано. Однако np.array(X_train).shape дает мне форму (100,). Мне бы хотелось, чтобы форма была (100, 128, 256, 1), чтобы я мог подавать ее в сверточный слой.

np.array(X_train)[0].shape выходы TensorShape([Dimension(128), Dimension(256), Dimension(1)]). Как мне сделать все это в один массив numpy с формой (100, 128, 256, 1)? В идеале вместо TensorShape это будет массив numpy.

Ниже находится load_image_train, который, как я считаю, должен вывести изображение:

@tf.function
def load_image_train(image):
  image = tf.expand_dims(image, -1)
  #image = tf.image.grayscale_to_rgb(image)

  input_image = tf.image.resize(image, (width, height))

  if tf.random.uniform(()) > 0.5:
    input_image = tf.image.flip_left_right(input_image)

  input_image = normalize(input_image)
  # print(input_image.shape)
  return input_image