axis_transform_5d
函数功能
把4D场景下的轴值转为5HD场景下的对应值。
函数原型
def axis_transform_5d(axis, data_format)
参数说明
参数 |
说明 |
|---|---|
axis |
轴值，int类型。 |
data_format |
数据format格式，字符串类型，目前仅支持"NCHW"和"NHWC"。如果是其他format，则直接返回原axis。 |
返回值说明
5HD场景下的对应轴值。
约束说明
data_format仅支持"NCHW"和"NHWC"。
调用示例
from tbe.common.utils import shape_util shape_util.axis_transform_5d(3, "NCHW")
把对应W对应的轴3转换为5HD场景下的对应值，即3。
父主题： shape相关工具
