sponge.function.squeeze_last_dim

sponge.function.squeeze_last_dim(input_x: Tensor)

返回从最后一个轴(axis=-1)删除大小为1的维度后的张量。

参数：

• input_x (Tensor) - 张量的shape为 $(x_1,x_2,...,x_R)$。

返回：

Tensor，张量的shape为 $(x_1,x_2,...,x_{R-1})$。

支持平台：

Ascend GPU CPU