cupy.stack#

cupy.stack(tup, axis=0, out=None, *, dtype=None, casting='same_kind')[source]#

沿新轴堆叠数组。

参数:
  • tup (数组序列) – 要堆叠的数组。

  • axis (int) – 堆叠数组所沿的轴。

  • out (cupy.ndarray) – 输出数组。

  • dtype (strdtype) – 如果提供,目标数组将具有此 dtype。不能与 out 同时提供。

  • casting ({‘no’, ‘equiv’, ‘safe’, ‘same_kind’, ‘unsafe’}, 可选) – 控制可能发生的数据类型转换种类。默认为 'same_kind'

返回:

堆叠后的数组。

返回类型:

cupy.ndarray

另请参阅

numpy.stack()