Матричная матрица Numpy Array

Question

У меня есть 3 массива -

XX, XY, YY, каждый из которых имеет форму (200, 300)

Я пытаюсь сложить массив формы 200x300x2x2

[XX, XY
XY, YY]

Как я могу это сделать?Спасибо!

Answer 1

Я бы сделал это так:

np.dstack((xx, xy, xy, yy)).reshape(200, 300, 2, 2)