nvidia.dali.experimental.dynamic.cat

nvidia.dali.experimental.dynamic.cat(input_0, /, *inputs, batch_size=None, device=None, axis=None, axis_name=None)

Joins the input tensors along an existing axis.

The shapes of the inputs must match in all dimensions except the concatenation axis.

Supported backends

• ‘cpu’

• ‘gpu’

Parameters:

• input_0 (Tensor/Batch) – Input to the operator.

• inputs[1..998] (Tensor/Batch, optional) – This function accepts up to 998 optional positional inputs

Keyword Arguments:

• axis (int, optional, default = 0) –

  Axis along which the input tensors are concatenated.

  Accepted range is [-ndim, ndim-1]. Negative indices are counted from the back.

• axis_name (str, optional) –

  Name of the axis along which the tensors are concatenated.

  This argument is mutually exclusive with axis. This argument requires that at least one input has a non-empty layout and that all non-empty input layouts match.