jax.numpy.hstack

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

按顺序水平堆叠数组（按列）。

LAX-backend 对 numpy.hstack() 的实现。

原始文档字符串如下。

这相当于沿着第二个轴进行连接，除了对于1维数组，它沿着第一个轴进行连接。重新构建由 hsplit 分割的数组。

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

参数:

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

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

返回:

stacked – 由给定数组堆叠形成的数组。

返回类型:

ndarray