shape_refine

函数功能

当reduce轴为None时，返回输入shape去除1后的shape列表；当reduce轴不为None时，返回输入去除1后的shape列表及排序后的reduce轴列表，格式为（shape,axis）的元组。

函数原型

def shape_refine(shape, reduce_axis=None, keep_dims=True):

参数说明

返回值说明

• 如果输入只有shape，则输出也只有shape。
• 如果输入同时有shape和reduce_axis，则输出是元组（shape,axis）。

约束说明

reduce轴对应的shape值为1，则不能进行优化。

调用示例

示例一：

from tbe.common.utils import shape_util 
shape_util.shape_refine((32, 64, 64, 1), reduce_axis=None) 

返回(32, 64, 64)。

示例二：

from tbe.common.utils import shape_util 
shape_util.shape_refine((32, 64, 64, 1), reduce_axis=[-2, -1], keep_dims=False) 

返回([32, 64, 64], [2])。