stack

dragon.stack(
  inputs,
  axis=0,
  **kwargs
)[source]

Stack the inputs along the given axis.

All the dimensions of inputs should be same.

Parameters:
  • inputs (Sequence[dragon.Tensor]) – The inputs.
  • axis (int, optional, default=0) – The axis to stack, can be negative.
Returns:

dragon.Tensor – The output tensor.