文档
注册
评分
提单
论坛
小AI

squeeze_shape

函数功能

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

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

函数原型

def squeeze_shape(shape)

参数说明

参数

说明

shape

需要去除dim为1的维度的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]。

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词