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)