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

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。

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

当前产品无相关内容

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