wrap_axes_to_positive

函数功能

将axis或者axis列表中的负值优化为正值。

例如axes：（1, 0, -2, -1），rank（维度）为4，则执行完wrap_axis_to_positive后，输出的axes为（1, 0, 2, 3）。

即：shape总共有4维； -1表示倒数第一个axis，即axis=3；-2表示倒数第二个axis，即axis=2。

函数原型

def wrap_axes_to_positive(axes, rank)

参数说明

返回值说明

返回将负值进行优化后的axes。

约束说明

无

调用示例

from tbe.common.utils import shape_util
axes = shape_util.wrap_axis_to_positive([-2, -1], 4)

执行完wrap_axis_to_positive后，axes=[2, 3]。