mindspore.ops.flatten

mindspore.ops.flatten(input, order='C', *, start_dim=1, end_dim=- 1)[源代码]

沿着从 start_dim 到 end_dim 的维度，对输入Tensor进行展平。

参数：

  • input (Tensor) - 输入Tensor。

  • order (str, 可选) - 仅支持 'C' 和 'F' 。 'C' 表示按行优先顺序 (C风格) 展平， 'F' 表示按列优先顺序 (Fortran风格) 展平。默认值： 'C' 。

关键字参数：

  • start_dim (int, 可选) - 要展平的第一个维度。默认值： 1 。

  • end_dim (int, 可选) - 要展平的最后一个维度。默认值： -1 。

返回：

Tensor。如果没有维度被展平，返回原始的输入 input，否则返回展平后的Tensor。如果 input 是零维Tensor，将会返回一个一维Tensor。

异常：

  • TypeError - input 不是Tensor。

  • TypeError - order 不是string类型。

  • ValueError - order 是string类型，但不是 'C' 或 'F' 。

  • TypeError - start_dim 或 end_dim 不是int类型。

  • ValueError - 规范化后， start_dim 大于 end_dim 。

  • ValueError - start_dim 或 end_dim 不在 [-input.dim, input.dim-1] 范围内。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> input_x = Tensor(np.ones(shape=[1, 2, 3, 4]), mindspore.float32)
>>> output = ops.flatten(input_x)
>>> print(output.shape)
(1, 24)