下载
中文
注册

enum class TensorFormat

说明

描述模型输入的数据排布。

定义

enum class TensorFormat {
        UNKNOWN = -1,
        NCHW = 0,
        NHWC = 1,
        ND = 2,
        NC1HWC0 = 3
     };

成员

成员名称

描述

UNKNOWN

未知的数据格式

NCHW

[批次大小,输入通道数,高,宽]

NHWC

[批次大小,高,宽,输入通道数]

ND

[批次大小,数据维度]

NC1HWC0

[批次大小,输入通道数1,高,宽,输入通道数0]