Python 合并不同形状的三维数组

所以我现在正在做的一个项目有一个问题，我有3个3D数组，形状（36,18,18），（18,36,18）和（18,18,36），我需要合并它们，只有一个形状数组（36,36,36）。有人知道怎么做吗？

根据输入数组类型应用相同的逻辑

arr = [
    [36, 18, 18],
    [18, 36, 18],
    [18, 18, 36]
    ]
new_array = []
for outer_index, outer_list in enumerate(arr):
    for ind in range(len(outer_list)):
        if ind == outer_index:
            new_array.append(outer_list[ind])
print(new_array)

输出：

[36, 36, 36]

---

你能详细说明你的问题吗？因为我不知道如何合并这三个矩阵，也就是说18x18x36x3元素，以创建一个新的36x36x36元素矩阵，因为知道元素的数量将不同。向我们展示如何将（4,2）与（2,4）合并（以生成（4,4））。请举个完整的例子。