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 |
需要优化的axis或者axis的列表。 |
rank |
需要优化的axes对应的shape的rank值。 如果axes中的值不在[-rank, rank)之间，则会抛出如下异常： RuntimeError "Axis must between [-%d, %d)." |
返回值说明
返回将负值进行优化后的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]。
父主题： shape相关工具
