Как я могу создать список функций узла в виде массива Numpy

Question

Для задачи машинного обучения на основе графика мне нужно ввести массив Numpy в следующем формате. На самом деле это массив массивов. Я пробовал несколько вариантов, таких как vstack, concatenate и т. Д. c. но у меня ничего не работает.

Это формат, который у меня сейчас есть.

array([[[0., 0., 0., ..., 0., 0., 0.],
        [0., 0., 0., ..., 0., 0., 0.],
        [0., 0., 0., ..., 0., 0., 0.]],

       [[0., 0., 0., ..., 0., 0., 0.],
        [0., 0., 0., ..., 0., 0., 0.],
        [0., 0., 0., ..., 0., 0., 0.]]], dtype=float32)

shape этого ввода:

(2, 3, 5)

Но мне нужно иметь массив NumPy с shape из:

(2,)

Это означает, что ввод должен быть массив массивов , как показано ниже.

array([array([[1., 0., 0., 0., 0.],
       [0., 1., 0., 0., 0.],
       [0., 0., 0., 0., 1.]], dtype=float32),
       array([[1., 0., 0., 0., 0.],
       [1., 0., 0., 0., 0.],
       [0., 0., 0., 0., 1.]], dtype=float32)], dtype=object)

Может кто-нибудь сказать мне, как преобразовать текущий вход (форма: (2, 3, 5)) в ожидаемый вход (форма: * 1028) *)?

Спасибо