sponge.function.concat_last_dim

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

Connect tensor in the last axis (axis=-1).

Connect input tensors along with the last 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.