mindspore.ops.flatten

mindspore.ops.flatten(input_x)[源代码]

扁平化(Flatten)输入Tensor,不改变0轴的size。

参数:

  • input_x (Tensor) - 待扁平化的Tensor,其shape为 \((N, \ldots)\)\(N\) 表示batch size。

返回:

Tensor,其shape为 \((N, X)\) 的Tensor,其中 \(X\) 是余下维度的乘积。

异常:

  • TypeError - input_x 不是Tensor。

  • ValueError - input_x 的shape长度小于1。

支持平台:

Ascend GPU CPU

样例:

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