np.stack

dstack方法的主要作用是将两个或更多的数组沿着第三个轴（深度方向）堆叠起来。

import numpy as np

# 创建两个二维数组（具有相同的行数和列数）
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])

# 使用dstack方法堆叠
result = np.dstack((a, b))

print(result)
# 输出：
# [[[1 5]
#  [2 6]]
# 
#  [[3 7]
#  [4 8]]]

import numpy as np

# 创建一个二维数组和一个一维数组（可以广播到二维）
a = np.array([[1, 2], [3, 4]])
b = np.array([5, 6])

# 使用dstack方法堆叠
result = np.dstack((a, b))

print(result)
# 输出：
# [[[1 5]
#  [2 6]]
# 
#  [[3 5]
#  [4 6]]]

参考：https://blog.csdn.net/2402_83361138/article/details/138047679