mindspore.ops.Flatten

class mindspore.ops.Flatten

扁平化（Flatten）输入Tensor，不改变0轴的size。

更多参考详见 mindspore.ops.flatten()。

输入：

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

输出：

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

支持平台：

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)
>>> flatten = ops.Flatten()
>>> output = flatten(input_x)
>>> print(output.shape)
(1, 24)