enum class TensorFormat
说明
描述模型输入的数据排布。
定义
enum class TensorFormat { UNKNOWN = -1, NCHW = 0, NHWC = 1, ND = 2, NC1HWC0 = 3 };
成员
成员名称 |
描述 |
---|---|
UNKNOWN |
未知的数据格式 |
NCHW |
[批次大小,输入通道数,高,宽] |
NHWC |
[批次大小,高,宽,输入通道数] |
ND |
[批次大小,数据维度] |
NC1HWC0 |
[批次大小,输入通道数1,高,宽,输入通道数0] |
父主题: 数据结构参考