check_and_init_5hdc_reduce_support

函数功能

当tensor的数据排布格式为5HD时，校验是否支持reduce功能。

函数原型

def check_and_init_5hdc_reduce_support(input_tensor, axis)

参数说明

返回值说明

如果支持，则返回True。否则，返回False。

约束说明

如果input_tensor是5HD，axis是1和4，数据类型为float16，但没有配置ori_shape或ori_format，则抛RuntimeError。

调用示例

from tbe.common.utils import para_check 
input_tensor = {}
input_tensor["format"] = "NC1HWC0"
input_tensor["dtype"] = "float16"
input_tensor["ori_shape"] = (32, 64, 64, 3)
input_tensor["ori_format"] = "NHWC"
para_check.check_and_init_5hdc_reduce_support(input_tensor, [1,4]) 

检查5HD的input_tensor是否支持reduce，返回True。