mindspore_lite.Format

class mindspore_lite.Format[源代码]

MindSpore Lite的“张量”类型。例如：格式。NCHW。

有关详细信息，请参见 Format 。运行以下命令导入包：

from mindspore_lite import 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`

用法

由于Python API中的 mindspore_lite.Tensor 是直接使用pybind11技术包装C++ API， Format 在Python API和C++ API之间有一对一的对应关系，修改 Format 的方法在 tensor 类的set和get方法中。
- set_format: 在 format_py_cxx_map 中以Python API中的 Format 为关键字进行查询，并获取C++ API中的 Format ，将其传递给C++ API中的 set_format 方法。
- get_format: 通过C++ API中的 get_format 方法在C++ API中获取 Format ，以C++ API中的 Format 为关键字在 format_cxx_py_map 中查询，返回在Python API中的 Format 。
以下是一个示例：

  from mindspore_lite import Format
  from mindspore_lite import Tensor

  tensor = Tensor()
  tensor.set_format(Format.NHWC)
  tensor_format = tensor.get_format()
  print(tensor_format)

运行结果如下：

.. code-block::

    Format.NHWC