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_formatdst_format 不是str。

  • TypeError - input_x 不是数据类型为int32的Tensor。

支持平台:

Ascend GPU CPU

样例:

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