mindspore.ops.Flatten

class mindspore.ops.Flatten[源代码]

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

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

支持平台:

Ascend GPU CPU

样例:

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