mindspore.ops.flatten

mindspore.ops.flatten(input_x)[source]

Flattens a tensor without changing its batch size on the 0-th axis.

Parameters

input_x (Tensor) – Tensor of shape \((N, \ldots)\) to be flattened, where \(N\) is batch size.

Returns

Tensor, the shape of the output tensor is \((N, X)\), where \(X\) is the product of the remaining dimension.

Raises

  • TypeError – If input_x is not a Tensor.

  • ValueError – If length of shape of input_x is less than 1.

Supported Platforms:

Ascend GPU CPU

Examples

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