sponge.function.squeeze_first_dim

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

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)\).