描述张量的组织形式。
1 2 3 4 5 6
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形式组织张量。