jax.numpy.dstack

jax.numpy.dstack(tup, dtype=None)

按深度顺序堆叠数组（沿第三轴）。

LAX 后端的 numpy.dstack() 实现。

原始文档字符串如下。

这相当于沿着第三个轴进行连接，在形状为 (M,N) 的二维数组被重塑为 (M,N,1) 和形状为 (N,) 的一维数组被重塑为 (1,N,1) 之后。重建由 dsplit 分割的数组。

此函数对于最多3维的数组最有意义。例如，对于具有高度（第一轴）、宽度（第二轴）和r/g/b通道（第三轴）的像素数据。函数 concatenate、stack 和 block 提供了更通用的堆叠和连接操作。

参数:

• tup (sequence of arrays) – 数组在除第三个轴以外的所有轴上必须具有相同的形状。1-D 或 2-D 数组必须具有相同的形状。

• dtype (DTypeLike | None)

返回:

stacked – 通过堆叠给定的数组形成的数组，至少是三维的。

返回类型:

ndarray