描述模型输入的数据排布。
enum class TensorFormat { UNKNOWN = -1, NCHW = 0, NHWC = 1, ND = 2, NC1HWC0 = 3 };
成员名称
描述
UNKNOWN
未知的数据格式
NCHW
[批次大小,输入通道数,高,宽]
NHWC
[批次大小,高,宽,输入通道数]
ND
[批次大小,数据维度]
NC1HWC0
[批次大小,输入通道数1,高,宽,输入通道数0]