Numpy ndarray с различными формами для изучения модели в Керасе

Question

У меня есть список массивов с разными формами.Я пытаюсь преобразовать этот список в массивный массив, чтобы создать пакетный образец для Keras.На выходе я хочу массив с формами (batch_size,?, 20), где '?'является переменной величиной.Я пытаюсь это сделать:

a = np.random.random((5,20))
b = np.random.random((2,20))
c = np.random.random((7,20))
d = [a,b,c]
np.array(d).shape
> (3,)

Когда я отправляю эту партию в Keras, у меня возникает следующая проблема:

ValueError: Error when checking input: expected Input_Dim to have 3 dimensions, but got array with shape (3, 1)

Answer 1

Может быть, этот простой пример может помочь:

import numpy as np

a = np.random.random((5,20))
b = np.random.random((2,20))
c = np.random.random((7,20))

d = np.array([a,b,c])
print(d.shape) # (3,)

d = d[np.newaxis]
print(d.shape) # (1, 3)