sponge.function.concat_first_dim

sponge.function.concat_first_dim(input_x: Tensor)[source]

Connect tensor in the first axis (axis=0).

Connect input tensors along with the first axis.

Parameters: input_x (Union[tuple, list]) – A tuple or a list of input tensors.
Returns: Tensor. A concatenated Tensor with the same type as input_x.