axis_check
函数功能
对轴的值进行合法性校验,并返回排好序且是正数的轴值,轴值按照升序进行排序。
函数原型
def axis_check(shape_len, axis):
参数说明
参数 |
说明 |
---|---|
shape_len |
shape大小 |
axis |
轴,可以是int类型的数值或者数组。 支持负数,代表倒数第几个轴。 axis必须在shape_len的轴数范围内,不能越界。 |
返回值说明
轴值,如果多个则排好序,如果是负数,则被转换为正数。
约束说明
无
调用示例
from tbe.common.utils import shape_util shape_util.axis_check(5, -1)
返回值:4。
父主题: shape相关工具