jax.numpy.vstack

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

按顺序垂直堆叠数组（按行）。

LAX-backend 实现的 numpy.vstack()。

原始文档字符串如下。

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

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

参数:

• tup (sequence of ndarrays) – 数组必须在除第一个轴以外的所有轴上具有相同的形状。一维数组必须具有相同的长度。

• dtype (str or dtype) – 如果提供，目标数组将具有此数据类型。不能与 out 一起提供。

返回:

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

返回类型:

ndarray