jax.numpy.column_stack

jax.numpy.column_stack(tup)

将一维数组堆叠为二维数组的列。

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

原始文档字符串如下。

将一系列一维数组堆叠为列，形成一个二维数组。二维数组按原样堆叠，就像使用 hstack 一样。一维数组首先被转换为二维列。

参数:

tup (sequence of 1-D or 2-D arrays.) – 要堆叠的数组。它们必须具有相同的第一维度。

返回:

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

返回类型:

2-D array