mindspore.ops.DataFormatVecPermute

class mindspore.ops.DataFormatVecPermute(src_format='NHWC', dst_format='NCHW')

将输入按从 src_format 到 dst_format 的变化重新排列。

参数：

• src_format (str, 可选) - 原先的数据排列格式，可以为’NHWC’和’NCHW’之一。默认值：’NHWC’。

• dst_format (str, 可选) - 目标数据排列格式，可以为’NHWC’和’NCHW’之一。默认值：’NCHW’。

输入：

• input_x (Tensor) - shape为(4, )或(4, 2)的输入Tensor。数据类型为int32或int64。

输出：

与 input_x 的shape和数据类型一致的Tensor。

异常：

• TypeError - 输入 input_x 不是Tensor。

• TypeError - input_x 的数据类型不是int32或int64。

• ValueError - input_x 的shape不为(4, )或(4, 2)。

• ValueError - src_format 或 dst_format 不是’NHWC’或’NCHW’之一。

支持平台：

GPU CPU

样例：

>>> class Net(nn.Cell):
...     def __init__(self, src_format="NHWC", dst_format="NCHW"):
...         super().__init__()
...         self.op = ops.DataFormatVecPermute(src_format, dst_format)
...     def construct(self, x):
...         return self.op(x)
...
>>> net = Net()
>>> x = Tensor(np.array([1, 2, 3, 4]).astype(np.int32))
>>> output = net(x)
>>> print(output)
[1 4 2 3]