Повторите массив по каналам

Question

Для массива формы (Нет примеров, строки, высоты, канала).Как я могу просто заменить каналы на Нет примеров?Я искал np.repeat(), но мне не удалось его применить.

import numpy as np
array = np.array([
                  [
                     [[0],[1]],
                     [[2],[3]],
                     [[4],[5]]
                  ],

                  [
                     [[0],[1]],
                     [[2],[3]],
                     [[4],[5]]
                  ],

                  [
                     [[0],[1]],
                     [[2],[3]],
                     [[4],[5]]
                  ],

                  [
                     [[0],[1]],
                     [[2],[3]],
                     [[4],[5]]
                  ]
                 ])
array.shape # (4, 3, 2, 1)

Мне нужен массив формы (4, 3, 2, 4).Каналы должны быть заменены количеством обучающих примеров.

Answer 1

Вы можете использовать np.tile:

np.tile(array, (1, 1, 1, array.shape[0]))

или np.repeat:

np.repeat(array[:, :, :,], array.shape[0], axis=3)