sponge.function.squeeze_last_dim

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

Return the Tensor after deleting the dimension of size 1 from the last axis (axis=-1).

Parameters: input_x (Tensor) – The shape of tensor is \((x_1, x_2, ..., x_R)\).
Returns: Tensor, the shape of tensor is \((x_1, x_2, ..., x_{R-1})\).