mindspore.ops.DataFormatDimMap

class mindspore.ops.DataFormatDimMap(src_format='NHWC', dst_format='NCHW')[源代码]

返回源数据格式中的目标数据格式的维度索引。

参数：

src_format (str) - 源数据格式中的可选值。格式可以是 'NHWC' 和 'NCHW' 。默认值： 'NHWC' 。
dst_format (str) - 目标数据格式中的可选值。格式可以是 'NHWC' 和 'NCHW' 。默认值： 'NCHW' 。

输入：

input_x (Tensor) - 输入Tensor，每个元素都用作源数据格式的维度索引。建议值在[-4, 4)范围内，仅支持int32。

输出：

输出Tensor，返回给定目标数据格式的维度索引，与 input_x 具有相同的数据类型和shape。

异常：

TypeError - src_format 或 dst_format 不是str。
TypeError - input_x 不是数据类型为int32的Tensor。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> from mindspore import Tensor, ops
>>> input_x = Tensor([0, 1, 2, 3], mindspore.int32)
>>> dfdm = ops.DataFormatDimMap()
>>> output = dfdm(input_x)
>>> print(output)
[0 3 1 2]