• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1mindspore.ops.DataFormatDimMap
2==============================
3
4.. py:class:: mindspore.ops.DataFormatDimMap(src_format='NHWC', dst_format='NCHW')
5
6    返回源数据格式中的目标数据格式的维度索引。
7
8    参数：
9        - **src_format** (str) - 源数据格式中的可选值。格式可以是 ``'NHWC'`` 和 ``'NCHW'`` 。默认值： ``'NHWC'`` 。
10        - **dst_format** (str) - 目标数据格式中的可选值。格式可以是 ``'NHWC'`` 和 ``'NCHW'`` 。默认值： ``'NCHW'`` 。
11
12    输入：
13        - **input_x** (Tensor) - 输入Tensor，每个元素都用作源数据格式的维度索引。建议值在[-4, 4)范围内，仅支持int32。
14
15    输出：
16        输出Tensor，返回给定目标数据格式的维度索引，与 `input_x` 具有相同的数据类型和shape。
17
18    异常：
19        - **TypeError** - `src_format` 或 `dst_format` 不是str。
20        - **TypeError** - `input_x` 不是数据类型为int32的Tensor。