get_tensor_format - 3.0 简体中文

获取运行器的张量格式。

原型设计

TensorFormat get_tensor_format();

参数

无

返回

TensorFormat：NHWC / HCHW

用法

auto format = runner->get_tensor_format();
switch (format) {
    case vart::Runner::TensorFormat::NCHW:
        // do something
        break;
    case vart::Runner::TensorFormat::NHWC:
        // do something
        break;
}