sponge.function.stack_first_dim

sponge.function.stack_first_dim(input_x: Tensor)

Stacks a list of tensors in the first axis (axis=0).

Parameters

input_x (Union[tuple, list]) – A Tuple or list of Tensor objects with the same shape and type.

Returns

Tensor. A stacked Tensor with the same type as input_x.

Supported Platforms:

Ascend GPU CPU