描述张量的组织形式。
enum TensorArrangementType {
TYPE_NHWC = 0,
TYPE_NCHW = 1,
TYPE_NHW = 2,
TYPE_NWH = 3
};
参数名 |
说明 |
|---|---|
TYPE_NHWC |
以NHWC形式组织张量。 |
TYPE_NCHW |
以NCHW形式组织张量。 |
TYPE_NHW |
以NHW形式组织张量。 |
TYPE_NWH |
以NWH形式组织张量。 |