squeeze_shape

函数功能

去除shape中dim值为1的维度，即去除shape中的值1。

例如，tensor a的shape为(1,2,3,4)，则调用完此接口后，tensor a的shape为(2,3,4)。

函数原型

def squeeze_shape(shape)

参数说明

返回值说明

返回去除dim为1的维度的shape。

约束说明

无。

调用示例

from tbe.common.utils import shape_util 
shape = shape_util.squeeze_shape([2,1,3,1,4]) 

执行完squeeze_shape后，shape=[2,3,4]。