一个np.stack的例子

import numpy as np

# 定义几个形状相同的数组
a = np.array([[1, 2, 3], [4, 5, 6]])
b = np.array([[7, 8, 9], [10, 11, 12]])
c = np.array([[13, 14, 15], [16, 17, 18]])
d = np.array([[19, 20, 21], [22, 23, 24]])

# 沿着不同的轴进行堆叠
stacked0 = np.stack((a, b, c, d), axis=0)
stacked1 = np.stack((a, b, c, d), axis=1)
stacked2 = np.stack((a, b, c, d), axis=2)

# 查看堆叠后的形状
print(stacked0.shape) # 输出: (4, 2, 3)
print(stacked1.shape) # 输出: (2, 4, 3)
print(stacked2.shape) # 输出: (2, 3, 4)