sponge.function.squeeze_penulti

查看源文件
sponge.function.squeeze_penulti(input_x: Tensor)[源代码]

返回从倒数第二个轴(axis=-2)删除大小为1的维度后的张量。

参数:
  • input_x (Tensor) - 张量的shape为 (x1,x2,...,xR1,xR)

返回:

Tensor,张量的shape为 (x1,x2,...,xR)

支持平台:

Ascend GPU CPU