cupy.stack#

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

Stacks arrays along a new axis.

Parameters:
  • tup (sequence of arrays) – Arrays to be stacked.

  • axis (int) – Axis along which the arrays are stacked.

  • out (cupy.ndarray) – Output array.

  • dtype (str or dtype) – If provided, the destination array will have this dtype. Cannot be provided together with out.

  • casting ({‘no’, ‘equiv’, ‘safe’, ‘same_kind’, ‘unsafe’}, optional) – Controls what kind of data casting may occur. Defaults to 'same_kind'.

Returns:

Stacked array.

Return type:

cupy.ndarray

See also

numpy.stack()