sponge.function.stack_first_dim

sponge.function.stack_first_dim(input_x: Tensor)

堆叠第一根轴上的张量列表(axis=0)。

参数：

• input_x (Union[tuple, list]) - 具有相同 shape 和类型的张量对象的元组或列表。

返回：

Tensor。与 input_x 类型相同的堆叠张量。

支持平台：

Ascend GPU CPU