sponge.function.stack_penulti

sponge.function.stack_penulti(input_x: Tensor)

Stacks a list of tensors in the penultimate axis (axis=-2).

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