Склейка в глубину в NumPy

Сложить плоские таблицы вдоль третьей оси позволяет функция dstack. Она склеивает двумерные массивы одинаковой формы вдоль новой оси: каждая исходная таблица становится «слой» внутри блока.

Две таблицы (2, 3) дают результат формы (2, 3, 2): по два слоя в каждой ячейке координат. Сложим две простые таблицы и выведем форму:

import numpy as np

front = np.array([[1, 2, 3], [4, 5, 6]])
back = np.array([[7, 8, 9], [10, 11, 12]])
layers = np.dstack((front, back))
print(np.shape(layers)) # выведет (2, 3, 2)

Сложите две таблицы (2, 2) из единиц и из двоек в глубину и выведите форму (2, 2, 2).

Сложите таблицы [[0, 1], [2, 3]] и [[4, 5], [6, 7]] в глубину и выведите число слоёв через последний размер формы.