mindspore_lite.Format

查看源文件
class mindspore_lite.Format[源代码]

Format 类定义MindSpore Lite中Tensor的格式。

目前,支持以下 Format :

定义

说明

Format.DEFAULT

默认格式。

Format.NCHW

按批次N、通道C、高度H和宽度W的顺序存储张量数据。

Format.NHWC

按批次N、高度H、宽度W和通道C的顺序存储张量数据。

Format.NHWC4

C轴4字节对齐格式的 Format.NHWC 。

Format.HWKC

按高度H、宽度W、核数K和通道C的顺序存储张量数据。

Format.HWCK

按高度H、宽度W、通道C和核数K的顺序存储张量数据。

Format.KCHW

按核数K、通道C、高度H和宽度W的顺序存储张量数据。

Format.CKHW

按通道C、核数K、高度H和宽度W的顺序存储张量数据。

Format.KHWC

按核数K、高度H、宽度W和通道C的顺序存储张量数据。

Format.CHWK

按通道C、高度H、宽度W和核数K的顺序存储张量数据。

Format.HW

按高度H和宽度W的顺序存储张量数据。

Format.HW4

w轴4字节对齐格式的 Format.HW 。

Format.NC

按批次N和通道C的顺序存储张量数据。

Format.NC4

C轴4字节对齐格式的 Format.NC 。

Format.NC4HW4

C轴4字节对齐和W轴4字节对齐格式的 Format.NCHW 。

Format.NCDHW

按批次N、通道C、深度D、高度H和宽度W的顺序存储张量数据。

Format.NWC

按批次N、宽度W和通道C的顺序存储张量数据。

Format.NCW

按批次N、通道C和宽度W的顺序存储张量数据。

Format.NDHWC

按批次N、深度D、高度H、宽度W和通道C的顺序存储张量数据。

Format.NC8HW8

C轴8字节对齐和W轴8字节对齐格式的 Format.NCHW 。

样例:

>>> # Method 1: Import mindspore_lite package
>>> import mindspore_lite as mslite
>>> print(mslite.Format.NHWC)
Format.NHWC
>>> # Method 2: from mindspore_lite package import Format
>>> from mindspore_lite import Format
>>> print(Format.NHWC)
Format.NHWC