jax.numpy.stack

jax.numpy.stack(arrays, axis=0, out=None, dtype=None)

沿新轴连接数组的序列。

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

原始文档字符串如下。

axis 参数指定新轴在结果维度中的索引。例如，如果 axis=0 它将是第一个维度，如果 axis=-1 它将是最后一个维度。

Added in version 1.10.0.

参数:

• arrays (sequence of array_like) – 每个数组必须具有相同的形状。

• axis (int, optional) – 结果数组中沿着输入数组堆叠的轴。

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

• out (None)

返回:

stacked – 堆叠数组比输入数组多一个维度。

返回类型:

ndarray