sponge.function.squeeze_first_dim

sponge.function.squeeze_first_dim(input_x: Tensor)

Return the Tensor after deleting the dimension of size 1 from the first axis (axis=0).

Parameters

input_x (Tensor) – The shape of tensor is $(x_1,x_2,...,x_R)$.

Returns

Tensor, the shape of tensor is $(x_2,...,x_R)$.

Supported Platforms:

Ascend GPU CPU